search

SharmaLlama / ticktack

An open-source carbon box model implementation built on JAX.
https://sharmallama.github.io/ticktack
MIT License
11 stars 3 forks source link

jax new version #52

Closed ahessl closed 7 months ago

ahessl commented 8 months ago

Jax has a new version. Can you check for conflicts?

benjaminpope commented 7 months ago

Hi @ahessl,

Sorry, I was travelling at a conference - I see there were some deprecated features (jax.config, and jnp.is1d) in the new version of jax that have now been fixed.

The new 1.1.4 version passes all the unit tests now - is it working for you?

Cheers,

Ben

ahessl commented 7 months ago

Yup! Thanks much!!