LICENSE
README.md
pyproject.toml
jaxonmodels/__init__.py
jaxonmodels.egg-info/PKG-INFO
jaxonmodels.egg-info/SOURCES.txt
jaxonmodels.egg-info/dependency_links.txt
jaxonmodels.egg-info/requires.txt
jaxonmodels.egg-info/top_level.txt
jaxonmodels/functions/__init__.py
jaxonmodels/functions/text.py
jaxonmodels/functions/utils.py
jaxonmodels/models/__init__.py
jaxonmodels/models/alexnet.py
jaxonmodels/models/clip.py
jaxonmodels/models/convnext.py
jaxonmodels/models/efficientnet.py
jaxonmodels/models/esm.py
jaxonmodels/models/mamba.py
jaxonmodels/models/resnet.py
jaxonmodels/models/swin_transformer.py
jaxonmodels/models/vit.py
jaxonmodels/statedict2pytree/__init__.py
jaxonmodels/statedict2pytree/model_orders.py
tests/test_convnext_layers.py
tests/test_esm_affine.py
tests/test_esm_layers.py
tests/test_swin_layers.py