piwheels

psgd-jax

An implementation of PSGD optimizer in JAX.

Installation

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

pip3 install psgd-jax

Dependencies

• matplotlib
• optax
• jax
• numpy

Releases

Version	Released	Bullseye Python 3.9	Bookworm Python 3.11	Trixie Python 3.13	Files
0.2.9	2024-12-31
psgd_jax-0.2.9-py3-none-any.whl (34 KB)
0.2.8	2024-12-08
psgd_jax-0.2.8-py3-none-any.whl (34 KB)
0.2.7	2024-11-10
psgd_jax-0.2.7-py3-none-any.whl (34 KB)
0.2.6	2024-11-07
psgd_jax-0.2.6-py3-none-any.whl (34 KB)
0.2.5	2024-10-29
psgd_jax-0.2.5-py3-none-any.whl (34 KB)
0.2.4	2024-10-23
psgd_jax-0.2.4-py3-none-any.whl (34 KB)
0.2.3	2024-10-07
psgd_jax-0.2.3-py3-none-any.whl (34 KB)
0.2.2	2024-10-07
psgd_jax-0.2.2-py3-none-any.whl (34 KB)
0.2.1	2024-10-03
psgd_jax-0.2.1-py3-none-any.whl (35 KB)
0.2.0	2024-10-02
psgd_jax-0.2.0-py3-none-any.whl (34 KB)
0.1.10	2024-10-01
psgd_jax-0.1.10-py3-none-any.whl (34 KB)
0.1.9	2024-10-01
psgd_jax-0.1.9-py3-none-any.whl (34 KB)
0.1.8	2024-10-01
psgd_jax-0.1.8-py3-none-any.whl (34 KB)
0.1.7	2024-09-29
psgd_jax-0.1.7-py3-none-any.whl (34 KB)
0.1.6	2024-09-29
psgd_jax-0.1.6-py3-none-any.whl (34 KB)
0.1.5	2024-08-22
psgd_jax-0.1.5-py3-none-any.whl (27 KB)
0.1.4	2024-08-16
psgd_jax-0.1.4-py3-none-any.whl (27 KB)
0.1.3	2024-08-11
psgd_jax-0.1.3-py3-none-any.whl (27 KB)
0.1.2	2024-08-10
psgd_jax-0.1.2-py3-none-any.whl (27 KB)
0.1.1	2024-08-07
psgd_jax-0.1.1-py3-none-any.whl (27 KB)
0.1.0	2024-08-07
psgd_jax-0.1.0-py3-none-any.whl (27 KB)

Issues with this package?

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