Go use cryojax ! It's awesome! I can't wait to see what people do with it.
Miro Astore
M​ichael J. O'Brien et al.: CryoJAX: a cryo-electron microscopy image-simulation library in JAX #CryoEM #ImageSimulation #AutomaticDifferentiation @harvard@unewhaven@Cornell...#IUCr https://journals.iucr.org/paper?S2059798326000550
The authors have developed cryoJAX, a cryo-EM image-simulation library for developing data-analysis techniques across cryo-EM modalities. CryoJAX is built on JAX, an emerging scientific computing framework in Python well suited for cryo-EM data analysis.