brentyi / jaxlieLinks
Rigid transforms + Lie groups for JAX
☆291Updated 4 months ago
Alternatives and similar repositories for jaxlie
Users that are interested in jaxlie are comparing it to the libraries listed below
Sorting:
- Sparse nonlinear least squares in JAX☆234Updated 3 months ago
- Learning with 3D rotations, a hitchhiker’s guide to SO(3) - ICML 2024☆253Updated last month
- 3D Gaussian Splatting in JAX☆64Updated last year
- Adjoint-Equivariant Neural Network for Semi-Simple Lie Algebras☆28Updated 8 months ago
- Bayesian filters in PyTorch☆157Updated 2 years ago
- Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…☆73Updated last year
- Pytrees + dataclasses ❤️☆68Updated 4 months ago
- A Smooth Representation of SO(3) for Deep Rotation Learning with Uncertainty.☆188Updated 11 months ago
- Scalable Differentiable Physics for Learning and Control (ICML2020)☆178Updated last year
- A differentiable physics engine for robotics☆351Updated 11 months ago
- Volumetric structures such as voxels and SDFs implemented in pytorch☆170Updated last week
- RoMa: A lightweight library to deal with 3D rotations in PyTorch.☆594Updated last month
- Python interface to Dojo☆67Updated 2 years ago
- A differentiable physics engine and multibody dynamics library for control and robot learning.☆147Updated 2 weeks ago
- ☆222Updated last year
- Implementation of the method proposed in the paper "Neural Descriptor Fields: SE(3)-Equivariant Object Representations for Manipulation"☆234Updated 3 years ago
- Differentiable simulation for system identification and visuomotor control☆197Updated last year
- Pure static Lie groups in Numpy, Pytorch, Jax, and C++☆36Updated 11 months ago
- Riemannian geometry in JAX☆14Updated 3 months ago
- Run PyTorch in JAX. 🤝☆283Updated last week
- PyTinyRenderer is a simple C++/CPU renderer, this fork adds Python bindings and triangle clipping and OpenGL compatible view/projection m…☆24Updated 10 months ago
- Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.☆67Updated last year
- Publication-ready visualization of 3D objects and point clouds in seconds.☆37Updated 5 months ago
- We are implementing differentiable models of robot manipulators, which allows us to learn typically assumed to be known models of robots …☆251Updated 2 years ago
- JAX port of Persistent Independent Particles☆36Updated 2 years ago
- ☆795Updated 3 months ago
- Use Jax functions in Pytorch☆249Updated 2 years ago
- Efficient Differentiable Simulation of Articulated Bodies (ICML2021)☆102Updated 3 years ago
- Repository containing the code for Implicit Manifold Gaussian Process.☆15Updated last year
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆57Updated last month