This is a library of some nice nonlinear systems on the GPU. It uses JAX to put the computations on the GPU. It's written in Hy.
pip install .The library provides tools for analyzing nonlinear dynamical systems using GPU acceleration.
Contributions are welcome. Please feel free to submit a pull request.
Q: Why are you using this stupid language (Hy) and not python?
A: Because I like it. Also, it's easier to write in a functional style in a lisp, and JAX needs pure functions. If it bothers you, use hy2py to produce the equivalent python code.
GPL v3.