search

mjo22 / cryojax

Cryo electron microscopy image simulation and analysis built on JAX.
https://mjo22.github.io/cryojax/
GNU Lesser General Public License v2.1
29 stars 9 forks source link

jax.tree_util -> jax.tree #267

Open mjo22 opened 2 weeks ago

mjo22 commented 2 weeks ago

cryojax must be updated for new JAX API