GitHub repository: patrick-kidger/esm2quinox
Primary language: Python
Description: An implementation of ESM2 in Equinox+JAX