Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
☆58Feb 15, 2026Updated 2 weeks ago
Alternatives and similar repositories for torch2jax
Users that are interested in torch2jax are comparing it to the libraries listed below
Sorting:
- Run PyTorch in JAX. 🤝☆313Oct 13, 2025Updated 4 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆132Feb 20, 2026Updated last week
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆54Updated this week
- Kinematic and dynamic models of continuum and articulated soft robots.☆15Nov 22, 2025Updated 3 months ago
- A metrics library for the JAX ecosystem☆40Mar 16, 2023Updated 2 years ago
- A dead simple Python package for creating custom JAX pytree objects☆66Nov 14, 2024Updated last year
- Official URDF and SDF models of the R1 humanoid robot.☆16Dec 6, 2023Updated 2 years ago
- A high-performance reinforcement learning library in jax specialized for robotic learning☆22Sep 4, 2023Updated 2 years ago
- ☆233Sep 3, 2023Updated 2 years ago
- Numerical quadrature with JAX☆77Feb 16, 2026Updated 2 weeks ago
- A Python-based Trajectory Optimization Framework☆22Oct 29, 2024Updated last year
- Automatic Functional Differentiation in JAX☆81Sep 18, 2025Updated 5 months ago
- Code and experiments for the NeurIPS 2023 paper Stabilized Neural Differential Equations for Learning Dynamics with Explicit Constraints☆12Mar 26, 2024Updated last year
- Probabilistic deep learning using JAX☆15Feb 8, 2025Updated last year
- Generator of binary installers for libraries built by vcpkg, alpha version.☆11Aug 5, 2021Updated 4 years ago
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆12May 8, 2023Updated 2 years ago
- ☆120Jan 27, 2026Updated last month
- Simple differentiable approximate ocean models built with JAX.☆16Oct 13, 2023Updated 2 years ago
- A CLI tool for generating Boost.Python/pybind11 bindings from C/C++☆15Aug 22, 2020Updated 5 years ago
- ☆13Dec 19, 2023Updated 2 years ago
- A simple Python package for deep learning using forward automatic differentiation based on JAX.☆14Aug 17, 2022Updated 3 years ago
- Julia interface for SNOPT7☆10Jun 16, 2021Updated 4 years ago
- Awesome openai gym environments☆12Aug 6, 2019Updated 6 years ago
- Cross-platform CMake/C++ library to get the installation prefix of shared library in a relocatable way.☆17Dec 1, 2024Updated last year
- ☆13Apr 7, 2025Updated 10 months ago
- Implementation of Computer Vision Models in JAX (equinox)☆18Jan 15, 2026Updated last month
- A Fast, Portable Deep Reinforcement Learning Library for Continuous Control☆13Jul 26, 2023Updated 2 years ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆546Feb 16, 2026Updated 2 weeks ago
- ☆219Feb 21, 2026Updated last week
- Files defining the Open Robotics dependencies☆13Updated this week
- AD backends for LogDensityProblems.jl.☆13Jan 1, 2026Updated 2 months ago
- Rapid prototyping of iCub applications using Python☆13Feb 20, 2026Updated last week
- Additional shapes to decorate MeshCat scenes (frames, text, ...)☆17Jun 30, 2025Updated 8 months ago
- Learning to optimize (L2O) package that provides basic functionalities to help fit proxy models for optimization.☆15Apr 1, 2025Updated 11 months ago
- ☆16May 30, 2023Updated 2 years ago
- E-book for AIS1003☆18Oct 27, 2023Updated 2 years ago
- Python Parser for SDFormat files.☆14Apr 5, 2022Updated 3 years ago
- Examples showing how to use SWIG to wrap MATLAB Coder generated C and C++ code for other languages☆11Apr 10, 2023Updated 2 years ago
- Differentiable trajectory optimization in Julia.☆65Nov 25, 2025Updated 3 months ago