cpgoodri / jax_transformations3d
Implementation of transformations.py that is JAX compatible
☆13Updated last year
Alternatives and similar repositories for jax_transformations3d:
Users that are interested in jax_transformations3d are comparing it to the libraries listed below
- Python package for creating interpolating splines (for position and rotation)☆65Updated 6 months ago
- Auto-differentiable and hardware-accelerated force density method☆88Updated last week
- Scalable Differentiable Physics for Learning and Control (ICML2020)☆171Updated 6 months ago
- Differentiable Cutting Simulator☆109Updated 4 months ago
- A differentiable physics engine and multibody dynamics library for control and robot learning.☆102Updated this week
- ☆210Updated last year
- ☆34Updated 3 years ago
- Differentiable interface to FEniCS for JAX☆52Updated 3 years ago
- This repository contains the source code to perform Geometry-aware Bayesian Optimization (GaBO) on Riemannian manifolds.☆50Updated 3 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆124Updated 5 months ago
- A set of custom deep network layers written in PyTorch to enable computation with unordered particle sets with an emphasis on fluid dynam…☆66Updated 4 years ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆93Updated last year
- Pytrees + dataclasses ❤️☆63Updated 7 months ago
- Rigid transforms + Lie groups in JAX☆241Updated 2 months ago
- We compare the gradients calculated by different differentiable contact model implementations.☆46Updated 2 years ago
- Differentiable simulation for system identification and visuomotor control☆188Updated last year
- GPU/TPU accelerated nonlinear least-squares curve fitting using JAX☆56Updated last year
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆45Updated 2 weeks ago
- PyTorch-FEniCS interface☆99Updated 3 years ago
- Python interface to Dojo☆65Updated 2 years ago
- Distance computation and collision detection in 3D.☆65Updated 2 months ago
- Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.☆67Updated last year
- Efficient Differentiable Simulation of Articulated Bodies (ICML2021)☆96Updated 3 years ago
- Python interface for CUDA implementation of OSQP☆89Updated 3 years ago
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆62Updated last month
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆58Updated last month
- Repository containing the code for Implicit Manifold Gaussian Process.☆13Updated 6 months ago
- Structured framework for learning mechanical systems in PyTorch☆25Updated 5 years ago
- Solver-in-the-Loop: Learning from Differentiable Physics to Interact with Iterative PDE-Solvers☆152Updated 2 years ago
- Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…☆70Updated last year