brentyi / jaxlie

Related projects: ⓘ

• brentyi / jaxls
  Sparse nonlinear least squares for JAX
  ☆171Updated this week
• martius-lab / hitchhiking-rotations
  Learning with 3D rotations, a hitchhiker’s guide to SO(3) - ICML 2024
  ☆200Updated last month
• yklcs / jaxsplat
  3D Gaussian Splatting in JAX
  ☆52Updated 3 months ago
• utiasSTARS / bingham-rotation-learning
  A Smooth Representation of SO(3) for Deep Rotation Learning with Uncertainty.
  ☆182Updated this week
• stanford-iprl-lab / torchfilter
  Bayesian filters in PyTorch
  ☆146Updated last year
• anthonysimeonov / ndf_robot
  Implementation of the method proposed in the paper "Neural Descriptor Fields: SE(3)-Equivariant Object Representations for Manipulation"
  ☆218Updated 2 years ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆60Updated 2 months ago
• princeton-vl / lietorch
  ☆673Updated 10 months ago
• google / trajax
  ☆192Updated last year
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆187Updated 10 months ago
• UM-ARM-Lab / pytorch_volumetric
  Volumetric structures such as voxels and SDFs implemented in pytorch
  ☆106Updated this week
• pimdh / relie
  Experiments for the AISTATS publication on Reparameterizing Distributions over Lie Groups
  ☆52Updated 2 years ago
• naver / roma
  RoMa: A lightweight library to deal with 3D rotations in PyTorch.
  ☆499Updated 2 months ago
• utiasSTARS / liegroups
  Python implementation of SO2, SE2, SO3, and SE3 matrix Lie groups using numpy or pytorch
  ☆328Updated 2 years ago
• vicariousinc / PGMax
  Loopy belief propagation for factor graphs on discrete variables, in JAX!
  ☆62Updated 2 months ago
• google-research / visu3d
  3d without friction (Torch, TF, Jax, Numpy)
  ☆145Updated 4 months ago
• JoeyTeng / jaxrenderer
  Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…
  ☆58Updated 10 months ago
• gradsim / gradsim
  Differentiable simulation for system identification and visuomotor control
  ☆169Updated 10 months ago
• facebookresearch / iSDF
  Real-time Neural Signed Distance Fields for Robot Perception
  ☆417Updated last year
• dojo-sim / Dojo.jl
  A differentiable physics engine for robotics
  ☆307Updated 5 months ago
• brentyi / dfgo
  Differentiable Factor Graph Optimization for Learning Smoothers @ IROS 2021
  ☆78Updated last year
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆224Updated last year
• nikihowe / myriad
  Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.
  ☆58Updated last year
• igilitschenski / deep_bingham
  Code for "Deep Orientaton Uncertainty Learning based on a Bingham Loss" (ICLR2020)
  ☆31Updated 2 months ago
• facebookresearch / baspacho
  Direct solver for sparse SPD matrices for nonlinear optimization. Implements supernodal Cholesky decomposition algorithm, and supports GP…
  ☆83Updated last year
• google-deepmind / einshape
  ☆97Updated 2 months ago
• geometric-kernels / GeometricKernels
  Geometric kernels on manifolds, meshes and graphs
  ☆213Updated last week
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆136Updated this week
• FlyingGiraffe / vnn
  Vector Neurons: A General Framework for SO(3)-Equivariant Networks
  ☆452Updated last year
• dojo-sim / dojopy
  Python interface to Dojo
  ☆59Updated last year