Skip to content

v1.3.0
Compare
Choose a tag to compare
@svwingerden svwingerden released this 24 Jan 14:53
· 4 commits to main since this release
v1.3.0
01f3257

What's Changed

  • Fix seeding across chains & in init_loss (NB seeding not backwards reproducible) by @svwingerden in #98
  • added an SGMCMC sampler, with prior and preconditioning. This subsumes the earlier SGLD and SGHNT implementations, and allows f.e. RMSProp as a sampler
  • deprecation warnings for SGLD and SGNHT (calling SGMCMC with the same arguments should work)
  • a small refactor of sampler requirements (backwards compatible)
  • cleaned up the requirements
  • Fixed small bugs with LLCs on TPU with BF16 and TPUv2/v3s
  • more tests, including snapshot tests so we know if we break something

Full Changelog: v1.2.0...v1.3.0

Contributors

svwingerden
Assets 6
Loading