
co-lead developer
Constrain gravitational potentials from stellar stream curvature
A JAX package that fits stream tracks with splines and compares their curvature directly to a potential's accelerations. Co-led with Sirui Wu.
3 more not shown.
11 more not shown.
16 more not shown.

co-lead developer
Constrain gravitational potentials from stellar stream curvature
A JAX package that fits stream tracks with splines and compares their curvature directly to a potential's accelerations. Co-led with Sirui Wu.

lead developer
Galactic dynamics in JAX
Orbit integration, potentials and stream generation — GPU-accelerated and fully differentiable.
maintainer
Multiple dispatch in Python
Multiple dispatch for Python, with type-hint-driven method resolution — the dispatch layer under quax and unxt.

lead developer
Coordinates in JAX
Vectors, frames and transformations — differentiable, and unit-aware via unxt.

lead developer
Units in JAX
Unit-aware quantities that survive jit, grad and vmap.
22 more not shown.
8 more not shown.
2 more not shown.