Optax is a gradient processing and optimization library for JAX. Optax is designed to facilitate research by providing building blocks that can be easily recombined in custom ways. The goals are to: - Provide simple, well-tested, efficient implementations of core components. - Improve research productivity by enabling to easily combine low level ingredients into custom optimisers (or other gradient processing components). - Accelerate adoption of new ideas by making it easy for anyone to contribute.
| Uploaded | Mon Mar 31 00:39:34 2025 |
| md5 checksum | bf3173d68e44d75f4603ca2ae7a37e65 |
| arch | x86_64 |
| build | py38h06a4308_0 |
| depends | absl-py >=0.7.1, chex >=0.1.5, jax >=0.1.55, jaxlib >=0.1.37, numpy >=1.20,<2, python >=3.8,<3.9.0a0, typing_extensions >=3.10 |
| license | Apache-2.0 |
| license_family | Apache |
| md5 | bf3173d68e44d75f4603ca2ae7a37e65 |
| name | optax |
| platform | linux |
| sha1 | fcf6ece8f0342ac3e2f647bd577acb6163a4699d |
| sha256 | 1322311a4c22bc37d6a6d8397fb71c540c6ec865a956a8e4fb6509f338dcf340 |
| size | 212883 |
| subdir | linux-64 |
| timestamp | 1670009566717 |
| version | 0.1.4 |