veros (pypi): The versatile ocean simulator, in pure Python, powered by JAX.
Keywords: oceanography|python|parallel|numpy|multi-core|geophysics|ocean-model|mpi4py|jax