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 | 1687c668212de771c20d9c6c63596368 |
| arch | x86_64 |
| build | py39h06a4308_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.9,<3.10.0a0, typing_extensions >=3.10 |
| license | Apache-2.0 |
| license_family | Apache |
| md5 | 1687c668212de771c20d9c6c63596368 |
| name | optax |
| platform | linux |
| sha1 | 1d207491137b27ece4890fd033f7b9c3a14625da |
| sha256 | dbbad7b959764e8c3256bcfe71cae24166bc12e322f125e3d222411d32739c70 |
| size | 213233 |
| subdir | linux-64 |
| timestamp | 1670009600947 |
| version | 0.1.4 |