brentyi / jaxlieLinks
Rigid transforms + Lie groups for JAX
☆317Updated 9 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☆265Updated this week
- Learning with 3D rotations, a hitchhiker’s guide to SO(3) - ICML 2024☆259Updated last month
- Adjoint-Equivariant Neural Network for Semi-Simple Lie Algebras☆32Updated 4 months ago
- 3D Gaussian Splatting in JAX☆69Updated last year
- Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…☆79Updated 2 years ago
- Bayesian filters in PyTorch☆164Updated 2 years ago
- Pytrees + dataclasses ❤️☆72Updated last month
- Volumetric structures such as voxels and SDFs implemented in pytorch☆184Updated 3 months ago
- A Smooth Representation of SO(3) for Deep Rotation Learning with Uncertainty.☆189Updated last year
- ☆232Updated 2 years ago
- Scalable Differentiable Physics for Learning and Control (ICML2020)☆184Updated last year
- A differentiable physics engine and multibody dynamics library for control and robot learning.☆167Updated last week
- RoMa: A lightweight library to deal with 3D rotations in PyTorch.☆624Updated 5 months ago
- Differentiable simulation for system identification and visuomotor control☆206Updated 2 years ago
- Python interface to Dojo☆70Updated 3 years ago
- Implementation of the method proposed in the paper "Neural Descriptor Fields: SE(3)-Equivariant Object Representations for Manipulation"☆237Updated 3 years ago
- Run PyTorch in JAX. 🤝☆311Updated 3 months ago
- Riemannian geometry in JAX☆16Updated 8 months ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆58Updated 6 months ago
- A differentiable physics engine for robotics☆366Updated last year
- Repository containing the code for Implicit Manifold Gaussian Process.☆16Updated last year
- Pure static Lie groups in Numpy, Pytorch, Jax, and C++☆40Updated 4 months ago
- PyTinyRenderer is a simple C++/CPU renderer, this fork adds Python bindings and triangle clipping and OpenGL compatible view/projection m…☆24Updated last year
- Publication-ready visualization of 3D objects and point clouds in seconds.☆39Updated this week
- export JAX to ONNX☆68Updated this week
- Differentiable QP solver in JAX☆127Updated last year
- ☆819Updated 8 months ago
- The analysis and implementation of SO3 log map, and its jacobians. The special focus on edge cases (angle close to 0 or \pi).☆30Updated 2 years ago
- Use Jax functions in Pytorch☆259Updated 2 years ago
- We are implementing differentiable models of robot manipulators, which allows us to learn typically assumed to be known models of robots …☆260Updated 2 years ago