PyRank

Optax Python Packages

Python packages with the GitHub topic optax. Sorted by relevance, with stars and monthly downloads.
bsc-quantic
tn4ml

Tensor Networks for Machine Learning

731 22 8
evanatyourservice
psgd-jax

An implementation of PSGD optimizer in JAX.

634 36 4
jqmc-project
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.

634 17 2
NITHISHM2410
flax-pilot

A Simplistic trainer for Flax

501 6 0
yonesuke
jaxfss

JAX/Flax implementation of finite-size scaling

138 16 0
hr0nix
optax-adan

An implementation of adan optimizer for optax

136 6 0
NITHISHM2410
flax-pilot-cpu

A Simplistic trainer for Flax

1 6 0