Awesome

jax-ekf

Lightweight EKF implementation in JAX. Supports arbitrary ⊞-manifolds.