enpasos / jax2onnxLinks
export JAX to ONNX - focus on flax nnx models
☆36Updated this week
Alternatives and similar repositories for jax2onnx
Users that are interested in jax2onnx are comparing it to the libraries listed below
Sorting:
- ☆216Updated last year
- A differentiable physics engine and multibody dynamics library for control and robot learning.☆134Updated last week
- GPU-friendly, auto-differentiable LQR solver with JAX.☆172Updated last month
- Differentiable QP solver in JAX☆106Updated 9 months ago
- Fast QP Solver in JAX☆15Updated 10 months ago
- Rex is a JAX-powered framework for sim-to-real robotics.☆39Updated last month
- Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.☆66Updated last year
- Transforms your CasADi functions into batchable JAX-compatible functions. By combining the power of CasADi with the flexibility of JAX, J…☆153Updated 4 months ago
- Riemannian geometry in JAX☆14Updated 2 months ago
- adam implements a collection of algorithms for calculating rigid-body dynamics in Jax, CasADi, PyTorch, and Numpy.☆179Updated last week
- A differentiable physics engine for robotics☆346Updated 9 months ago
- Python interface to Dojo☆67Updated 2 years ago
- Stanford PhD Thesis☆18Updated 2 years ago
- Unified contact simulaton and collision detection☆42Updated 2 years ago
- Differentiable collision detection for capsules in JAX.☆47Updated last year
- ☆61Updated this week
- Learning Optimal Policies Through Contact in Differentiable Simulation☆105Updated last year
- ☆274Updated 5 months ago
- Conic Augmented Lagrangian Interior-Point SOlver☆72Updated 2 years ago
- improved Cross Entropy Method for trajectory optimization☆77Updated 3 years ago
- Mathematical Programming in JAX☆87Updated last month
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆230Updated last month
- A PyTorch implementation of TTGO algorithm and the applications presented in the paper "Tensor Train for Global Optimization Problems in …☆19Updated 8 months ago
- A differentiable simulator for robotic systems.☆31Updated 3 years ago
- Sampling-based model predictive control on GPU with JAX/MJX☆181Updated last month
- ☆102Updated last week
- On-the-fly conversions between Jax and NumPy tensors☆52Updated 2 years ago
- A GPU accelerated library for computing rigid body dynamics with analytical gradients☆113Updated 2 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆55Updated 2 months ago
- Dual purpose Acrobot and Pendubot Platform☆44Updated 2 weeks ago