LICENSE
README.md
pyproject.toml
setup.cfg
setup.py
jaxtra/__init__.py
jaxtra/activations.py
jaxtra/data_loader.py
jaxtra/layers.py
jaxtra/models.py
jaxtra.egg-info/PKG-INFO
jaxtra.egg-info/SOURCES.txt
jaxtra.egg-info/dependency_links.txt
jaxtra.egg-info/requires.txt
jaxtra.egg-info/top_level.txt