f0uriest / interpax

Interpolation and function approximation with JAX
MIT License
137 stars 11 forks source link

jaxlib version #11

Closed joglekara closed 1 year ago

joglekara commented 1 year ago

Great work that fills a clear need IMO!!

Any thoughts on bumping jax/jaxlib to 0.4.20?

joglekara commented 1 year ago

perhaps we can do a >= like in https://github.com/patrick-kidger/equinox/blob/main/pyproject.toml

Happy to send in a tested PR

f0uriest commented 1 year ago

I'm fine with bumping it, but I prefer to keep a cap (<=) on the max version since JAX is still in v0.x.y so no guaranteed API consistency.

allen-adastra commented 1 year ago

https://github.com/f0uriest/interpax/pull/12 fixed this.