yklcs / jaxsplat
3D Gaussian Splatting in JAX
β52Updated 3 months ago
Related projects: β
- Run PyTorch in JAX. π€β187Updated 10 months ago
- Neural Optimal Transport with Lagrangian Costsβ37Updated 2 months ago
- Pytrees + dataclasses β€οΈβ60Updated 2 months ago
- Rigid transforms + Lie groups in JAXβ223Updated 2 months ago
- Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrendeβ¦β58Updated 10 months ago
- Volumetric structures such as voxels and SDFs implemented in pytorchβ106Updated this week
- Flow-matching algorithms in JAXβ62Updated last month
- β40Updated 2 months ago
- Visualizations of the theory behind diffusion models.β63Updated 5 months ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.β34Updated last month
- Simple code for generating a color-coded latex table from raw dataβ146Updated 10 months ago
- Differentiable and accelerated spherical transforms with JAXβ123Updated this week
- π A powerful library for efficient training of Neural Fields at scale.β26Updated 7 months ago
- Use Jax functions in Pytorchβ224Updated last year
- Pytorch-like dataloaders in JAX.β52Updated last month
- Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"β39Updated last month
- JAX Arrays for human consumptionβ87Updated last year
- code for "Riemannian Flow Matching on General Geometries".β166Updated 6 months ago
- Learning with 3D rotations, a hitchhikerβs guide to SO(3) - ICML 2024β200Updated last month
- β97Updated 2 months ago
- Source code for Fathony, Sahu, Willmott, & Kolter, "Multiplicative Filter Networks", ICLR 2021.β90Updated 3 years ago
- Official JAX implementation of neural isometries - taming transformations for equivariant MLβ21Updated last month
- Sparse nonlinear least squares for JAXβ171Updated this week
- [NeurIPS 2021] Code for Learning Signal-Agnostic Manifolds of Neural Fieldsβ65Updated last year
- Diffusion models in PyTorchβ81Updated last week
- β145Updated last month
- β33Updated 2 years ago
- PyTorch implementation of the Hessian-free optimizerβ30Updated 3 months ago
- Code for the paper: Rotating Features for Object Discoveryβ46Updated last month
- Graph neural networks in JAX.β66Updated 3 months ago