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:33 2025 |
| md5 checksum | fd7ee06cc5593d37fbd8026cfe1b074a |
| arch | x86_64 |
| build | py310h06a4308_0 |
| depends | absl-py >=0.7.1, chex >=0.1.5, jax >=0.1.55, jaxlib >=0.1.37, numpy >=1.21,<2, python >=3.10,<3.11.0a0, typing_extensions >=3.10 |
| license | Apache-2.0 |
| license_family | Apache |
| md5 | fd7ee06cc5593d37fbd8026cfe1b074a |
| name | optax |
| platform | linux |
| sha1 | 8f03c4de5a2c3d7989e4629a5ecdd1c910a5072d |
| sha256 | 5542032b2759373495d359fda24462916742043061376fe989fe2b6995129acb |
| size | 214636 |
| subdir | linux-64 |
| timestamp | 1670009662821 |
| version | 0.1.4 |