brentyi / jaxlie

Related projects ⓘ

Alternatives and complementary repositories for jaxlie

• brentyi / jaxls
  Sparse nonlinear least squares in JAX
  ☆180Updated 2 weeks ago
• martius-lab / hitchhiking-rotations
  Learning with 3D rotations, a hitchhiker’s guide to SO(3) - ICML 2024
  ☆215Updated 3 months ago
• yklcs / jaxsplat
  3D Gaussian Splatting in JAX
  ☆54Updated 5 months ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆64Updated 4 months ago
• utiasSTARS / bingham-rotation-learning
  A Smooth Representation of SO(3) for Deep Rotation Learning with Uncertainty.
  ☆184Updated 2 months ago
• stanford-iprl-lab / torchfilter
  Bayesian filters in PyTorch
  ☆149Updated last year
• princeton-vl / lietorch
  ☆688Updated last year
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆200Updated last year
• google / trajax
  ☆200Updated last year
• vicariousinc / PGMax
  Loopy belief propagation for factor graphs on discrete variables, in JAX!
  ☆64Updated last month
• naver / roma
  RoMa: A lightweight library to deal with 3D rotations in PyTorch.
  ☆515Updated 3 weeks ago
• JoeyTeng / jaxrenderer
  Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…
  ☆65Updated last year
• UM-ARM-Lab / pytorch_volumetric
  Volumetric structures such as voxels and SDFs implemented in pytorch
  ☆121Updated 3 weeks ago
• anthonysimeonov / ndf_robot
  Implementation of the method proposed in the paper "Neural Descriptor Fields: SE(3)-Equivariant Object Representations for Manipulation"
  ☆219Updated 2 years ago
• YilingQiao / diffsim
  Scalable Differentiable Physics for Learning and Control (ICML2020)
  ☆165Updated 3 months ago
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆228Updated last year
• metaopt / optree
  OpTree: Optimized PyTree Utilities
  ☆152Updated this week
• nikihowe / myriad
  Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.
  ☆61Updated last year
• google-deepmind / einshape
  ☆99Updated 4 months ago
• google-research / visu3d
  3d without friction (Torch, TF, Jax, Numpy)
  ☆147Updated this week
• ami-iit / jaxsim
  A differentiable physics engine and multibody dynamics library for control and robot learning.
  ☆83Updated this week
• pimdh / relie
  Experiments for the AISTATS publication on Reparameterizing Distributions over Lie Groups
  ☆52Updated 2 years ago
• utiasSTARS / liegroups
  Python implementation of SO2, SE2, SO3, and SE3 matrix Lie groups using numpy or pytorch
  ☆335Updated 2 years ago
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆88Updated last year
• gradsim / gradsim
  Differentiable simulation for system identification and visuomotor control
  ☆181Updated last year
• facebookresearch / iSDF
  Real-time Neural Signed Distance Fields for Robot Perception
  ☆423Updated last year
• geometric-kernels / GeometricKernels
  Geometric kernels on manifolds, meshes and graphs
  ☆219Updated this week
• astro-informatics / s2fft
  Differentiable and accelerated spherical transforms with JAX
  ☆136Updated this week
• facebookresearch / baspacho
  Direct solver for sparse SPD matrices for nonlinear optimization. Implements supernodal Cholesky decomposition algorithm, and supports GP…
  ☆85Updated last year
• danielkelshaw / riemax
  Riemannian geometry in JAX
  ☆13Updated 3 months ago