piwheels

optax

A gradient processing and optimization library in JAX.

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install optax

Dependencies

• numpy
• chex
• absl-py
• jax
• jaxlib

Releases

Version	Released	Bullseye Python 3.9	Bookworm Python 3.11	Files
0.2.5	2025-06-10
optax-0.2.5-py3-none-any.whl (346 KB)
0.2.4	2024-11-12
optax-0.2.4-py3-none-any.whl (312 KB)
0.2.3	2024-07-09
optax-0.2.3-py3-none-any.whl (283 KB)
0.2.2	2024-03-27
optax-0.2.2-py3-none-any.whl (218 KB)
0.2.1	2024-03-07
optax-0.2.1-py3-none-any.whl (205 KB)
0.2.0	2024-03-06
optax-0.2.0-py3-none-any.whl (205 KB)
0.1.9	2024-02-03
optax-0.1.9-py3-none-any.whl (193 KB)
0.1.8	2024-01-15
optax-0.1.8-py3-none-any.whl (195 KB)
0.1.7	2023-07-26
optax-0.1.7-py3-none-any.whl (151 KB)
0.1.5	2023-04-20
optax-0.1.5-py3-none-any.whl (161 KB)
0.1.4	2022-11-21
optax-0.1.4-py3-none-any.whl (151 KB)
0.1.3	2022-07-13
optax-0.1.3-py3-none-any.whl (142 KB)
0.1.2	2022-04-13
optax-0.1.2-py3-none-any.whl (137 KB)
0.1.1	2022-02-04
optax-0.1.1-py3-none-any.whl (133 KB)
0.1.0	2021-11-18
optax-0.1.0-py3-none-any.whl (124 KB)
0.0.91	2021-11-18
optax-0.0.91-py3-none-any.whl (124 KB)
0.0.9	2021-07-02
optax-0.0.9-py3-none-any.whl (116 KB)
0.0.8	2021-06-10
optax-0.0.8-py3-none-any.whl (111 KB)
0.0.6	2021-03-26
optax-0.0.6-py3-none-any.whl (94 KB)
0.0.5	2021-03-23
optax-0.0.5-py3-none-any.whl (88 KB)
0.0.3	2021-03-16
optax-0.0.3-py3-none-any.whl (72 KB)
0.0.2	2020-12-04
optax-0.0.2-py3-none-any.whl (53 KB)
0.0.1	2020-08-14
optax-0.0.1-py3-none-any.whl (48 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue