facebookresearch / optimizer-plugins
A demonstration of source code transformation to implement automatic differentiation, compatible with an operation overload style AD library DiffKt.
☆13Updated 2 years ago
Alternatives and similar repositories for optimizer-plugins:
Users that are interested in optimizer-plugins are comparing it to the libraries listed below
- A framework for automatic differentiation in Kotlin☆60Updated 2 years ago
- Loop Nest - Linear algebra compiler and code generator.☆22Updated 2 years ago
- ☆21Updated this week
- Code for "Recognizing Scenes from Novel Viewpoints"☆29Updated 2 years ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆42Updated last year
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆17Updated last year
- Generative cellular automaton-like learning environments for RL.☆19Updated 2 months ago
- A LinearOperator implementation for PyTorch☆18Updated 4 years ago
- ☆11Updated 10 months ago
- Convolutions and more as einsum for PyTorch☆14Updated 9 months ago
- Contact-Aware Symplectic Integrator Network☆12Updated 2 years ago
- ☆42Updated last week
- ☆26Updated 9 months ago
- High-performance tokenized language data-loader for Python C++ extension☆12Updated 8 months ago
- Repo reproducing experimental results in "Addressing the Topological Defects of Disentanglement"☆22Updated 2 years ago
- ☆33Updated 4 years ago
- Authors implementation of LieTransformer: Equivariant Self-Attention for Lie Groups☆36Updated 4 years ago
- Neural Fixed-Point Acceleration for Convex Optimization☆29Updated 2 years ago
- 3D Gaussian Splatting in JAX☆57Updated 10 months ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆27Updated 3 years ago
- ☆45Updated 10 months ago
- Julia implementation of SMPL family of 3D human models.☆15Updated 3 months ago
- ☆9Updated 3 weeks ago
- Implementation of a holodeck, written in Pytorch☆17Updated last year
- flexible meta-learning in jax☆12Updated last year
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Implementation of Bayesian Sum-Product Networks☆13Updated 4 years ago
- ☆9Updated 4 years ago
- Hessian trace estimation using PyTorch and Hutch++☆19Updated 4 years ago
- ☆15Updated last year