1 result found Sort:

evanatyourservice
psgd_jax evanatyourservice
2
30
cc-by-4.0
4
Implementation of PSGD optimizer in JAX
jax psgd optax optimization preconditioning second-order-optimization
Created 2024-07-13
71 commits to main branch, last one 2 months ago