jebastin-nadar / jax-models Public

Notifications You must be signed in to change notification settings
Fork 0
Star 1

JAX/Flax model architectures compatible with timm

Apache-2.0 license

1 star 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 5 Commits
jam		jam
LICENSE		LICENSE
README.md		README.md

Repository files navigation

JAX Models

Common image encoders written in JAX/Flax. The architectures are similar to timm's implementations.

About

JAX/Flax model architectures compatible with timm

Apache-2.0 license

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%