CITATION
CITATION.cff
LICENSE
MANIFEST.in
NOTICE
README.md
pyproject.toml
setup.py
torchtangents/__init__.py
torchtangents/convert.py
torchtangents/nn.py
torchtangents/predict.py
torchtangents/py.typed
torchtangents/stax.py
torchtangents.egg-info/PKG-INFO
torchtangents.egg-info/SOURCES.txt
torchtangents.egg-info/dependency_links.txt
torchtangents.egg-info/requires.txt
torchtangents.egg-info/top_level.txt
torchtangents/_src/__init__.py
torchtangents/_src/convert.py
torchtangents/_src/nn/__init__.py
torchtangents/_src/nn/modules.py
torchtangents/_src/stax/__init__.py
torchtangents/_src/stax/batching.py
torchtangents/_src/stax/branching.py
torchtangents/_src/stax/combinators.py
torchtangents/_src/stax/elementwise.py
torchtangents/_src/stax/linear.py
torchtangents/_src/stax/pooling.py
torchtangents/_src/stax/requirements.py
torchtangents/_src/utils/__init__.py
torchtangents/_src/utils/dataclasses.py
torchtangents/_src/utils/kernel.py
torchtangents/_src/utils/typing.py
torchtangents/_src/utils/utils.py