| Name | Modified | Size | Downloads / Week |
|---|---|---|---|
| Parent folder | |||
| neural-tangents-0.6.0.tar.gz | 2022-07-18 | 227.3 kB | |
| neural_tangents-0.6.0-py2.py3-none-any.whl | 2022-07-18 | 241.9 kB | |
| README.md | 2022-07-18 | 2.4 kB | |
| v0.6.0 source code.tar.gz | 2022-07-18 | 4.4 MB | |
| v0.6.0 source code.zip | 2022-07-18 | 4.4 MB | |
| Totals: 5 Items | 9.3 MB | 0 | |
New features:
* nt.empirical:
* New implementation=3 for nt.empirical, allowing to often speed-up or reduce the memory of the empirical NTK by orders of magnitude. Please see our ICML2022 paper Fast Finite Width Neural Tangent Kernel, new empirical NTK examples, and visit us on Thursday at ICML in-person!
* New experimental prototype of using our empirical NTK implementations in Tensorflow via nt.experimental.empirical_ntk_fn_tf.
* Make nt.empircial work with arbitrary pytrees.
nt.stax:
Improvements: * Slightly lower memory usage in batching. * Many improvements to documentation and type annotations. * Simplify test specifications and avoid relying on JAX testing utilities.
Bugfixes:
* Make nt.batch use the correct (local) number of devices in a multi-host setting (thanks @jglaser).