Rishit-dagli / torch-diffsimLinks
⚙️ minimal paralllelizable physics simulator supporting differentiation entirely in torch
☆33Updated last week
Alternatives and similar repositories for torch-diffsim
Users that are interested in torch-diffsim are comparing it to the libraries listed below
Sorting:
- 3D Gaussian Splatting in JAX☆65Updated last year
- Neural Optimal Transport with Lagrangian Costs☆60Updated 5 months ago
- Flow-matching algorithms in JAX☆106Updated last year
- Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…☆76Updated last year
- ☆120Updated 4 months ago
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆25Updated 9 months ago
- ☆35Updated 3 years ago
- Pytrees + dataclasses ❤️☆67Updated 6 months ago
- PyTinyRenderer is a simple C++/CPU renderer, this fork adds Python bindings and triangle clipping and OpenGL compatible view/projection m…☆24Updated last year
- Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"☆28Updated last year
- PyTorch implementation of Shortcut Models [Frans, 2025] with little modification☆58Updated 3 months ago
- Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling" [NeurIPS 2025]☆172Updated 2 weeks ago
- [NeurIPS 2024] Official implementation of "NeuralClothSim: Neural Deformation Fields Meet the Thin Shell Theory"☆36Updated last year
- Implementation of Danijar's latest iteration for his Dreamer line of work☆88Updated this week
- ☆44Updated 2 months ago
- Official repository for the paper "Flow Equivariant Recurrent Neural Networks"☆20Updated 4 months ago
- Volumetric structures such as voxels and SDFs implemented in pytorch☆175Updated 2 weeks ago
- Official JAX implementation of neural isometries - taming transformations for equivariant ML☆36Updated 3 months ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆38Updated last year
- Gradient-based constrained optimization for JAX☆35Updated 3 years ago
- Official PyTorch and JAX Implementation of "Harmonics of Learning: Universal Fourier Features Emerge in Invariant Networks"☆38Updated last year
- Rigid transforms + Lie groups for JAX☆305Updated 6 months ago
- Learning with 3D rotations, a hitchhiker’s guide to SO(3) - ICML 2024☆254Updated 3 months ago
- Generative cellular automaton-like learning environments for RL.☆19Updated 9 months ago
- NF-Layers for constructing neural functionals.☆90Updated last year
- Fast singularity detection with kernel☆37Updated last year
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆58Updated 3 months ago
- Code of the Paper "Time-Efficient Reinforcement Learning with Stochastic Stateful Policies"☆25Updated last year
- Run PyTorch in JAX. 🤝☆305Updated 3 weeks ago
- MoMo: Momentum Models for Adaptive Learning Rates☆19Updated last year