SWAG-jax. JAX implementation of Stochastic weight averaging-Gaussian by Maddox et al.

github.com/sdatkinson/SWAG-jax
by @sdatkinson

Problem

Audience

Keep exploring

Updates

No founder updates yet. Check back soon.