Alternatives and similar repositories for jaxls:

Users that are interested in jaxls are comparing it to the libraries listed below

• brentyi / jaxlie
  Rigid transforms + Lie groups for JAX
  ☆258Updated 2 weeks ago
• stanford-iprl-lab / torchfilter
  Bayesian filters in PyTorch
  ☆152Updated last year
• utiasSTARS / bingham-rotation-learning
  A Smooth Representation of SO(3) for Deep Rotation Learning with Uncertainty.
  ☆186Updated 7 months ago
• yklcs / jaxsplat
  3D Gaussian Splatting in JAX
  ☆57Updated 11 months ago
• UMich-CURLY / LieNeurons
  Adjoint-Equivariant Neural Network for Semi-Simple Lie Algebras
  ☆28Updated 4 months ago
• facebookresearch / baspacho
  Direct solver for sparse SPD matrices for nonlinear optimization. Implements supernodal Cholesky decomposition algorithm, and supports GP…
  ☆90Updated last year
• princeton-vl / lietorch
  ☆761Updated last week
• martius-lab / hitchhiking-rotations
  Learning with 3D rotations, a hitchhiker’s guide to SO(3) - ICML 2024
  ☆245Updated 4 months ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆62Updated 2 weeks ago
• brentyi / dfgo
  Differentiable Factor Graph Optimization for Learning Smoothers @ IROS 2021
  ☆87Updated 2 years ago
• vicariousinc / PGMax
  Loopy belief propagation for factor graphs on discrete variables, in JAX!
  ☆65Updated 7 months ago
• utiasSTARS / liegroups
  Python implementation of SO2, SE2, SO3, and SE3 matrix Lie groups using numpy or pytorch
  ☆347Updated last month
• google / trajax
  ☆212Updated last year
• david-m-rosen / Optimization
  A set of lightweight header-only template functions implementing commonly-used optimization methods on Riemannian manifolds and convex sp…
  ☆103Updated last year
• tu-rbo / differentiable-particle-filters
  Source code and data for the paper "Differentiable Particle Filters: End-to-End Learning with Algorithmic Priors"
  ☆185Updated last year
• pimdh / relie
  Experiments for the AISTATS publication on Reparameterizing Distributions over Lie Groups
  ☆52Updated 2 years ago
• google-research / visu3d
  3d without friction (Torch, TF, Jax, Numpy)
  ☆151Updated 5 months ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆240Updated 2 months ago
• google-deepmind / einshape
  ☆103Updated 10 months ago
• naver / roma
  RoMa: A lightweight library to deal with 3D rotations in PyTorch.
  ☆575Updated 2 months ago
• nurlanov-zh / so3_log_map
  The analysis and implementation of SO3 log map, and its jacobians. The special focus on edge cases (angle close to 0 or \pi).
  ☆25Updated last year
• patrick-kidger / optimistix
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆410Updated 2 weeks ago
• jlblancoc / tutorial-se3-manifold
  LaTeX sources of the technical report "A tutorial on SE(3) transformation parameterizations and on-manifold optimization"
  ☆186Updated 3 years ago
• anthonysimeonov / ndf_robot
  Implementation of the method proposed in the paper "Neural Descriptor Fields: SE(3)-Equivariant Object Representations for Manipulation"
  ☆232Updated 2 years ago
• google-deepmind / PGMax
  Loopy belief propagation for factor graphs on discrete variables in JAX
  ☆150Updated 6 months ago
• igilitschenski / deep_bingham
  Code for "Deep Orientaton Uncertainty Learning based on a Bingham Loss" (ICLR2020)
  ☆34Updated 10 months ago
• decargroup / pymlg
  Pure static Lie groups in Numpy, Pytorch, Jax, and C++
  ☆30Updated 7 months ago
• geometric-kernels / GeometricKernels
  Geometric kernels on manifolds, meshes and graphs
  ☆248Updated last month
• xflash96 / pybind11_package_example
  Example python package with pybind11 cpp extension
  ☆57Updated 4 years ago
• AaltoML / BayesNewton
  Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.
  ☆231Updated last year