Revision 115ddcbe73af08da7e7c5e6f414bc647f2f620d4 authored by John QiangZhang on 02 August 2022, 17:44:59 UTC, committed by jax authors on 09 August 2022, 21:30:06 UTC
Quick update: * clean the function_alias parameters. * disable custom_gradient and run through jax test. PiperOrigin-RevId: 464830257
1 parent 6d9512a
| File | Mode | Size |
|---|---|---|
| jax_cpp | ||
| __init__.py | -rw-r--r-- | 575 bytes |
| advi.py | -rw-r--r-- | 4.8 KB |
| datasets.py | -rw-r--r-- | 3.1 KB |
| differentially_private_sgd.py | -rw-r--r-- | 8.9 KB |
| examples_test.py | -rw-r--r-- | 2.0 KB |
| gaussian_process_regression.py | -rw-r--r-- | 4.5 KB |
| kernel_lsq.py | -rw-r--r-- | 2.5 KB |
| mnist_classifier.py | -rw-r--r-- | 3.1 KB |
| mnist_classifier_fromscratch.py | -rw-r--r-- | 3.0 KB |
| mnist_vae.py | -rw-r--r-- | 4.9 KB |
| onnx2xla.py | -rw-r--r-- | 4.7 KB |
| spmd_mnist_classifier_fromscratch.py | -rw-r--r-- | 4.9 KB |
Computing file changes ...
