Path Lines of Code CODE_OF_CONDUCT.md 57 COMPILE_README.md 47 CONTRIBUTING.md 2 README.md 264 examples/compilation/README.md 3 examples/compilation/eager_fusion.py 37 examples/compilation/fuse_module.py 41 examples/compilation/linear_train.py 60 examples/compilation/simple_function.py 21 examples/dp_cifar10/cifar10_opacus.py 336 examples/dp_cifar10/cifar10_transforms.py 362 examples/ensembling/parallel_train.py 62 examples/lennard_jones/lennard_jones.py 44 examples/maml_omniglot/README.md 11 examples/maml_omniglot/maml-omniglot-higher.py 164 examples/maml_omniglot/maml-omniglot-ptonly.py 164 examples/maml_omniglot/maml-omniglot-transforms.py 170 examples/maml_omniglot/support/omniglot_loaders.py 190 examples/maml_regression/evjang.py 84 examples/maml_regression/evjang_transforms.py 89 examples/maml_regression/evjang_transforms_module.py 91 notebooks/ensembling.ipynb 308 notebooks/jacobians_hessians.ipynb 479 notebooks/per_sample_grads.ipynb 360 test/pytest.ini 2 writing_batching_rules.md 72