LICENSE
MANIFEST.in
README.md
pyproject.toml
jaxnn/__init__.py
jaxnn/cli.py
jaxnn/py.typed
jaxnn/version.py
jaxnn.egg-info/PKG-INFO
jaxnn.egg-info/SOURCES.txt
jaxnn.egg-info/dependency_links.txt
jaxnn.egg-info/entry_points.txt
jaxnn.egg-info/requires.txt
jaxnn.egg-info/top_level.txt
jaxnn/data/__init__.py
jaxnn/data/config.py
jaxnn/data/constants.py
jaxnn/data/transforms.py
jaxnn/data/transforms_factory.py
jaxnn/layers/__init__.py
jaxnn/layers/activations.py
jaxnn/layers/attention.py
jaxnn/layers/attention_pool.py
jaxnn/layers/blur_pool.py
jaxnn/layers/create_act.py
jaxnn/layers/create_norm.py
jaxnn/layers/diff_attention.py
jaxnn/layers/drop.py
jaxnn/layers/drop_path.py
jaxnn/layers/format.py
jaxnn/layers/grn.py
jaxnn/layers/helpers.py
jaxnn/layers/identity.py
jaxnn/layers/layer_scale.py
jaxnn/layers/mlp.py
jaxnn/layers/norm.py
jaxnn/layers/patch_dropout.py
jaxnn/layers/patch_embed.py
jaxnn/layers/pool2d.py
jaxnn/layers/pos_embed_sincos.py
jaxnn/layers/trace_utils.py
jaxnn/models/__init__.py
jaxnn/models/_builder.py
jaxnn/models/_factory.py
jaxnn/models/_features.py
jaxnn/models/_helpers.py
jaxnn/models/_hub.py
jaxnn/models/_pretrained.py
jaxnn/models/_registry.py
jaxnn/models/_types.py
jaxnn/models/resnet.py
jaxnn/models/vision_transformer.py
tests/test_attention_pool.py
tests/test_builder.py
tests/test_grn.py
tests/test_model_create.py
tests/test_registry.py