LICENSE
MANIFEST.in
NOTICE
README.md
pyproject.toml
docs/_static/vmap/image_classifier.batched.txt
docs/_static/vmap/image_classifier.single.txt
docs/_static/vmap/recommendation.batched.txt
docs/_static/vmap/recommendation.single.txt
examples/autodiff.py
examples/per_example_gradients.py
examples/vmap.py
src/onnax/__init__.py
src/onnax/_diagnostics.py
src/onnax/_functional.py
src/onnax/_inference.py
src/onnax/_rule_registry.py
src/onnax/py.typed
src/onnax.egg-info/PKG-INFO
src/onnax.egg-info/SOURCES.txt
src/onnax.egg-info/dependency_links.txt
src/onnax.egg-info/requires.txt
src/onnax.egg-info/top_level.txt
src/onnax/_autodiff/__init__.py
src/onnax/_autodiff/_common.py
src/onnax/_autodiff/_context.py
src/onnax/_autodiff/_elementwise.py
src/onnax/_autodiff/_interface.py
src/onnax/_autodiff/_layout.py
src/onnax/_autodiff/_linalg.py
src/onnax/_autodiff/_reductions.py
src/onnax/_autodiff/_registry.py
src/onnax/_autodiff/_transform.py
src/onnax/_batching/__init__.py
src/onnax/_batching/_context.py
src/onnax/_batching/_elementwise.py
src/onnax/_batching/_indexing.py
src/onnax/_batching/_layout.py
src/onnax/_batching/_linalg.py
src/onnax/_batching/_nn.py
src/onnax/_batching/_reductions.py
src/onnax/_batching/_registry.py
src/onnax/_batching/_transform.py
tests/_testing.py
tests/test_autodiff.py
tests/test_batching.py
tests/test_batching_examples.py
tests/test_diagnostics.py
tests/test_documentation.py
tests/test_functional.py
tests/test_higher_order.py
tests/test_inference.py
tests/test_integration.py
tests/test_modern_opsets.py
tests/test_per_example_gradients.py
tests/test_rule_registry.py
tools/check_distribution.py