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 | d9334bf94a1b9c71f4638a008ad2fa0b |
| arch | x86_64 |
| build | py311h06a4308_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.11,<3.12.0a0, typing_extensions >=3.10 |
| license | Apache-2.0 |
| license_family | Apache |
| md5 | d9334bf94a1b9c71f4638a008ad2fa0b |
| name | optax |
| platform | linux |
| sha1 | 14a3b283fff3127947810f607364ad7d45a3ecff |
| sha256 | 60e81f083007a1be66e76ce106b0cd6286025b288f4159968817fbe58a6f00e3 |
| size | 275553 |
| subdir | linux-64 |
| timestamp | 1676860145693 |
| version | 0.1.4 |