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 3 weeks ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆42Updated last year
- A simple hypernetwork implementation in jax using haiku.☆23Updated 2 years ago
- Authors implementation of LieTransformer: Equivariant Self-Attention for Lie Groups☆36Updated 4 years ago
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆17Updated last year
- code associated with paper "Sparse Bayesian Optimization"☆26Updated last year
- Code for "Recognizing Scenes from Novel Viewpoints"☆29Updated 2 years ago
- Julia implementation of SMPL family of 3D human models.☆16Updated 4 months ago
- flexible meta-learning in jax☆13Updated last year
- ☆15Updated 2 years ago
- Explorations into the proposed Streaming Deep Reinforcement Learning, from University of Alberta☆19Updated 5 months ago
- We simulate a wind tunnel, place a rectangular occlusion in it, and then use gradient descent to turn the occlusion into a wing.☆25Updated 4 years ago
- ☆30Updated last year
- Texture mapping with variational auto-encoders☆40Updated 3 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆51Updated 2 weeks ago
- ☆26Updated 10 months ago
- PyTinyRenderer is a simple C++/CPU renderer, this fork adds Python bindings and triangle clipping and OpenGL compatible view/projection m…☆24Updated 6 months ago
- ☆33Updated 4 years ago
- Source-to-Source Debuggable Derivatives in Pure Python☆15Updated last year
- A LinearOperator implementation for PyTorch☆18Updated 4 years ago
- ☆26Updated 9 months ago
- 3D Gaussian Splatting in JAX☆57Updated 10 months ago
- ☆26Updated this week
- Better bindings for Python☆17Updated 2 years ago
- Contact-Aware Symplectic Integrator Network☆13Updated 2 years ago
- High quality implementations of imitation and inverse reinforcement learning algorithms☆14Updated 3 weeks ago
- ☆43Updated last month
- Code for Canonical 3D Deformable Mapping (C3DM) paper☆19Updated 3 years ago