Maximilian Stölzle

@armantekinalp Thank you for your answer and for your referral to the RL project. In case I decide to implement this functionality, I will try to merge it back to...

> I also run into the issue Timeout waiting for IOPub output for some of my student submissions. I haven't looked into it in enough detail to find out what...

> No, I haven't looked any further into this. I actually fixed it last night. In case you encounter the same / a similar issue in the future again: Opposite...

> What version of JAX and what version of NumPy are you using? I am using python 3.11, numpy 1.23.5, jax 0.4.14, equinox 0.10.11, diffrax 0.4.1

@cgarciae Could you please have a look at the pull requests? The package is quickly becoming unusable because of deprecations in Jax / Flax / optax.