"""Local flax_models copied from OneScience for this standalone package."""