Revision 3e5466c2f9c21edfad8a7ea1e973aec1a763d64c authored by jax authors on 18 November 2021, 17:19:17 UTC, committed by jax authors on 18 November 2021, 17:32:09 UTC
PiperOrigin-RevId: 410822358
1 parent 52421da
| File | Mode | Size |
|---|---|---|
| _src | ||
| example_libraries | ||
| experimental | ||
| image | ||
| interpreters | ||
| lax | ||
| lib | ||
| nn | ||
| numpy | ||
| ops | ||
| scipy | ||
| tools | ||
| __init__.py | -rw-r--r-- | 4.4 KB |
| abstract_arrays.py | -rw-r--r-- | 773 bytes |
| ad_checkpoint.py | -rw-r--r-- | 726 bytes |
| api_util.py | -rw-r--r-- | 784 bytes |
| cloud_tpu_init.py | -rw-r--r-- | 648 bytes |
| config.py | -rw-r--r-- | 697 bytes |
| core.py | -rw-r--r-- | 73.9 KB |
| custom_derivatives.py | -rw-r--r-- | 1.0 KB |
| distributed.py | -rw-r--r-- | 641 bytes |
| dlpack.py | -rw-r--r-- | 668 bytes |
| dtypes.py | -rw-r--r-- | 1.0 KB |
| errors.py | -rw-r--r-- | 984 bytes |
| flatten_util.py | -rw-r--r-- | 644 bytes |
| jaxpr_util.py | -rw-r--r-- | 3.7 KB |
| linear_util.py | -rw-r--r-- | 10.1 KB |
| prng.py | -rw-r--r-- | 884 bytes |
| profiler.py | -rw-r--r-- | 1.1 KB |
| py.typed | -rw-r--r-- | 0 bytes |
| random.py | -rw-r--r-- | 3.7 KB |
| test_util.py | -rw-r--r-- | 989 bytes |
| tree_util.py | -rw-r--r-- | 2.3 KB |
| util.py | -rw-r--r-- | 977 bytes |
| version.py | -rw-r--r-- | 634 bytes |
Computing file changes ...
