A small library of utilities to explore Classical Mechanics, heavily inspired by scmutils presented in the book Structure and Interpretation of Classical Mechanics by Sussman and Wisdom, and implemented in JAX, taking advantage of the awesome function transformations/compositions!
To get a quick overview, checkout the examples/ folder (for instance the simple pendulum example), or visit In search of Hidden Conservation Laws to see a cool demo of what the library is capable of!