Welcome to jaxkuramoto!

jaxkuramoto is a Python library for simulating the Kuramoto model, which has the following differential equation:

\[ \frac{\mathrm{d}\theta_{i}}{\mathrm{d} t}= \omega_{i} + \frac{K}{N}\sum_{j=1}^{N}\sin(\theta_{j}-\theta_{i}). \]

It is built on top of JAX, a library for differentiable programming in Python. It is designed to be easy to use and to be compatible with JAX’s JIT compilation and GPU acceleration.

You can view the source code on GitHub.

Getting Started

• Installation
• Synchronization
• Why Kuramoto model?
• How to use odeint
• How to use distribution

Examples

• Kuramoto model
• Bifurcation diagram
• Ott-Antonsen reduction
• Sakaguchi-Kuramoto model: Adding Phase Shifts to the Kuramoto Model
• Clustering and slow switching phenomena

Guides

• List of ODE solvers
• List of nonlinear solvers
• List of distributions
• List of predefined models

Package Reference

• Package Reference