qax

qax copied to clipboard

"module 'jax.core' has no attribute 'new_main'" using jax>=0.4.36

4

Hello! Thanks for this library, I've been using it for a while (via Lorax), but I now get an error after updating to the latest Jax version. Apparently Jax 0.4.36...

bminixhofer