Sparse nonlinear least squares in JAX
☆272Feb 26, 2026Updated last week
Alternatives and similar repositories for jaxls
Users that are interested in jaxls are comparing it to the libraries listed below
Sorting:
- Rigid transforms + Lie groups for JAX☆322Apr 24, 2025Updated 10 months ago
- Differentiable Factor Graph Optimization @ IROS 2021☆94Feb 22, 2023Updated 3 years ago
- Pytrees + dataclasses ❤️☆75Dec 19, 2025Updated 2 months ago
- Extending JAX with custom C++ and CUDA code☆403Aug 18, 2024Updated last year
- Learn symforce together :)☆32Aug 23, 2022Updated 3 years ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,028Dec 17, 2025Updated 2 months ago
- Bayesian filters in PyTorch☆165May 25, 2023Updated 2 years ago
- ☆233Sep 3, 2023Updated 2 years ago
- Generic EKF, with support for non-Euclidean manifolds☆25Apr 6, 2022Updated 3 years ago
- A Pytree Module system for Deep Learning in JAX☆213Feb 26, 2023Updated 3 years ago
- A Python-based Trajectory Optimization Framework☆22Oct 29, 2024Updated last year
- New version: https://github.com/chungmin99/pyroki☆19Apr 13, 2025Updated 10 months ago
- Tools for JAX☆50Feb 20, 2026Updated 2 weeks ago
- A library for differentiable nonlinear optimization☆2,006Jan 16, 2025Updated last year
- Python implementation of SO2, SE2, SO3, and SE3 matrix Lie groups using numpy or pytorch☆358Apr 1, 2025Updated 11 months ago
- A differentiable physics engine and multibody dynamics library for control and robot learning.☆179Updated this week
- Transforms your CasADi functions into batchable JAX-compatible functions. By combining the power of CasADi with the flexibility of JAX, J…☆176Feb 23, 2025Updated last year
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆241Dec 22, 2023Updated 2 years ago
- Exponential families for JAX☆76Feb 20, 2026Updated 2 weeks ago
- Fast symbolic computation, code generation, and nonlinear optimization for robotics☆1,587Feb 12, 2026Updated 3 weeks ago
- Factored inference for discrete-continuous smoothing and mapping.☆155Feb 12, 2025Updated last year
- A general and flexible factor graph non-linear least square optimization framework☆510Jan 14, 2022Updated 4 years ago
- Inference on non-linear dynamical systems written in JAX☆11Aug 20, 2020Updated 5 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆133Feb 20, 2026Updated 2 weeks ago
- Automatic Differentiated Extended Kalman Filter (ADEKF) - This is a generic EKF Implementation that uses automatic differentiation to get…☆125Jan 12, 2026Updated last month
- A small collection of Kalman Filters on Lie groups☆383Apr 13, 2023Updated 2 years ago
- A library for differentiable robotics on manifolds.☆1,510Dec 21, 2025Updated 2 months ago
- Code of paper "Associating Uncertainty to Extended Poses for on Lie Group IMU Preintegration with Rotating Earth"☆94Aug 7, 2020Updated 5 years ago
- Python parser for URDFs☆267Jan 23, 2026Updated last month
- ☆822May 10, 2025Updated 9 months ago
- adam implements a collection of algorithms for calculating rigid-body dynamics in Jax, CasADi, PyTorch, and Numpy.☆212Updated this week
- A Python package for probabilistic state space modeling with JAX☆932Jan 6, 2026Updated 2 months ago
- Mathematical operations for JAX pytrees☆209Dec 5, 2024Updated last year
- iDynFor☆15Mar 14, 2024Updated last year
- Dive into Jax, Flax, XLA and C++☆32Apr 1, 2020Updated 5 years ago
- 3D Gaussian Splatting in JAX☆72May 30, 2024Updated last year
- Differentiable collision detection for capsules in JAX.☆52Dec 2, 2023Updated 2 years ago
- Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…☆79Nov 14, 2023Updated 2 years ago
- Pytorch Implementation of TinyMPC☆63Jan 17, 2024Updated 2 years ago