yklcs / jaxsplat
3D Gaussian Splatting in JAX
β55Updated 7 months ago
Alternatives and similar repositories for jaxsplat:
Users that are interested in jaxsplat are comparing it to the libraries listed below
- Run PyTorch in JAX. π€β216Updated 3 weeks ago
- Neural Optimal Transport with Lagrangian Costsβ50Updated 6 months ago
- Flow-matching algorithms in JAXβ83Updated 5 months ago
- Rigid transforms + Lie groups in JAXβ240Updated 2 months ago
- Pytrees + dataclasses β€οΈβ63Updated 6 months ago
- Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrendeβ¦β68Updated last year
- Official JAX implementation of neural isometries - taming transformations for equivariant MLβ34Updated 5 months ago
- Visualizations of the theory behind diffusion models.β77Updated 9 months ago
- Learning with 3D rotations, a hitchhikerβs guide to SO(3) - ICML 2024β230Updated last month
- π A powerful library for efficient training of Neural Fields at scale.β28Updated last year
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).β17Updated 5 months ago
- β40Updated 2 months ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.β44Updated this week
- Differentiable and accelerated spherical transforms with JAXβ143Updated this week
- Use Jax functions in Pytorchβ232Updated last year
- Efficiently Composable Data Augmentation on the GPU with Jaxβ32Updated 6 months ago
- Sparse nonlinear least squares in JAXβ184Updated last month
- 3d without friction (Torch, TF, Jax, Numpy)β150Updated 2 months ago
- β153Updated 5 months ago
- Volumetric structures such as voxels and SDFs implemented in pytorchβ133Updated this week
- Pytorch-like dataloaders in JAX.β70Updated 3 months ago
- Canonical Factors for Hybrid Neural Fields @ ICCV 2023β102Updated last week
- Learning about Neural Radiance Fieldsβ67Updated 2 years ago
- [NeurIPS 2021] Code for Learning Signal-Agnostic Manifolds of Neural Fieldsβ67Updated last year
- JAX Arrays for human consumptionβ90Updated last year
- Geometry processing utilities compatible with jax for autodifferentiation.β86Updated last year
- VectorAdam for Rotation Equivariant Geometry Optimizationβ112Updated 2 years ago
- Stainless neural networks in JAXβ32Updated 3 months ago
- [NeurIPS 2024] Official implementation of "NeuralClothSim: Neural Deformation Fields Meet the Thin Shell Theory"β29Updated 3 months ago
- β79Updated 3 years ago