PyPI Stats

Search Packages

Find Python packages by name, description, GitHub topic, or filter by metrics
evanatyourservice
psgd-jax

Implementation of PSGD optimizer in JAX

1K 35 3
kousuke-nakano
jqmc

No description available

997 15 1
NITHISHM2410
flax-pilot

A Simplistic trainer for Flax

549 6 0
bsc-quantic
tn4ml

Tensor Networks for Machine Learning

250 23 7
hr0nix
optax-adan

An implementation of adan optimization algorithm for optax.

122 6 0
yonesuke
jaxfss

JAX/Flax implementation of finite-size scaling

121 15 0
NITHISHM2410
flax-pilot-cpu

A Simplistic trainer for Flax

1 6 0