jaxkuramoto (pypi): JAX implementation of Kuramoto model.
Keywords: jax|kuramoto-model|physics|python