Update oryx version to 0.2.7 and only expose `optax_schedule_free` when it is available.

jax-ml / bayeux

State of the art inference for your bayesian models.

https://jax-ml.github.io/bayeux/

Apache License 2.0

177 stars 8 forks source link

Closed copybara-service[bot] closed 6 months ago

copybara-service[bot] commented 6 months ago

Update oryx version to 0.2.7 and only expose optax_schedule_free when it is available.

Fixes #47