piwheels

jax-unirep

A performant and user-friendly reimplementation of UniRep in JAX.

Installation

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

pip3 install jax-unirep

Dependencies

• jaxlib
• optax
• tqdm
• scikit-learn
• multipledispatch
• numpy
• equinox
• optuna
• jax

Releases

Version	Released	Bookworm Python 3.11	Trixie Python 3.13	Files
3.0.0	2026-09-04
jax_unirep-3.0.0-py3-none-any.whl (73 MB)
2.2.0	2022-08-15
2.1.0	2021-10-05
jax_unirep-2.1.0-py3-none-any.whl (38 KB)
2.0.2	2021-05-09
2.0.1	2021-01-03
2.0.0	2020-12-14
1.0.3	2020-09-28
1.0.2	2020-09-14
1.0.1	2020-08-29
1.0	2020-08-13

Issues with this package?

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