jax-ml / bayeux

State of the art inference for your bayesian models.
https://jax-ml.github.io/bayeux/
Apache License 2.0
162 stars 6 forks source link

Add schedule free Adam optimizer. #46

Closed copybara-service[bot] closed 5 months ago

copybara-service[bot] commented 5 months ago

Add schedule free Adam optimizer.