Closed mjo22 closed 4 months ago
Whether or not this is added to cryojax
, to finish this PR I would like to add an inference tutorial notebook into the documentation for a cross-correlation based grid search.
Finished the tutorial for the grid search and ready to merge. This PR now also includes the new API, so merging will mark a big version release.
Addresses #201. The API is very generic and assumes nothing about cryo-EM. It could be released to the JAX community, operating at a similar level of abstraction as something like
optax
. I think it will be really useful for everyone to have this as a part of a maintained package, but my dilemma is that I am a little hesitant right now to maintain something for a more general community outside of cryo-EM. So I'm not sure yet if its appropriate to add this intocryojax
and keep it there, make it an independent package, or add it now and later make it an independent package.Aside from this, this PR also includes important packaging updates. The submodules in
cryojax
were getting a little superfluous. For example, I've eliminatedcryojax.core
and absorbedcryojax.io
intocryojax.data
.