Closed joglekara closed 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
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.
https://github.com/f0uriest/interpax/pull/12 fixed this.
Great work that fills a clear need IMO!!
Any thoughts on bumping jax/jaxlib to 0.4.20?