Skip to content

psgd-jax 0.2.1
Compare
Choose a tag to compare
@evanatyourservice evanatyourservice released this 03 Oct 04:12
· 18 commits to main since this release
v0.2.1
1a5e305

What's Changed

  • add min ndim triangular arg to better catch bias and scale params for diag preconditioners without catching linear params like (512, 1)
Assets 2
Loading