PyRank

Optax Python Packages

Python packages with the GitHub topic optax. Sorted by relevance, with stars and monthly downloads.
kousuke-nakano
jqmc

jQMC code implements two real-space ab initio quantum Monte Carlo (QMC) methods. Variatioinal Monte Carlo (VMC) and lattice regularized diffusion Monte Carlo (LRDMC) methods. jQMC achieves high-performance computations especially on GPUs.

981 16 2
evanatyourservice
psgd-jax

Implementation of PSGD optimizer in JAX

638 35 3
NITHISHM2410
flax-pilot

A Simplistic trainer for Flax

446 6 0
bsc-quantic
tn4ml

Tensor Networks for Machine Learning

277 22 8
hr0nix
optax-adan

An implementation of adan optimization algorithm for optax.

118 6 0
yonesuke
jaxfss

JAX/Flax implementation of finite-size scaling

112 15 0
NITHISHM2410
flax-pilot-cpu

A Simplistic trainer for Flax

1 6 0