LICENSE
README.md
pyproject.toml
setup.py
src/paxjaxlib/__init__.py
src/paxjaxlib/activations.py
src/paxjaxlib/core.py
src/paxjaxlib/initializers.py
src/paxjaxlib/layers.py
src/paxjaxlib/losses.py
src/paxjaxlib/metrics.py
src/paxjaxlib/models.py
src/paxjaxlib/regularizers.py
src/paxjaxlib/schedules.py
src/paxjaxlib/torch_bridge.py
src/paxjaxlib/training.py
src/paxjaxlib.egg-info/PKG-INFO
src/paxjaxlib.egg-info/SOURCES.txt
src/paxjaxlib.egg-info/dependency_links.txt
src/paxjaxlib.egg-info/requires.txt
src/paxjaxlib.egg-info/top_level.txt
tests/test_activations.py
tests/test_initializers.py
tests/test_layers.py
tests/test_losses.py
tests/test_metrics.py
tests/test_models.py
tests/test_new_layers.py
tests/test_regularizers.py
tests/test_torch_bridge.py