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
- Loop Nest - Linear algebra compiler and code generator.☆22Updated 2 years ago
- 🎲 A Kotlin DSL for probabilistic programming.☆12Updated 2 years ago
- A framework for automatic differentiation in Kotlin☆60Updated 2 years ago
- Authors implementation of LieTransformer: Equivariant Self-Attention for Lie Groups☆36Updated 3 years ago
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆17Updated 10 months ago
- ☆22Updated last month
- Code for "Journey to the BAOAB-limit: finding effective MCMC samplers for score-based models". See more at https://ajayj.com/journey.☆12Updated 2 years ago
- ☆12Updated last year
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆42Updated last year
- Computing the greatest common divisor with transformers, source code for the paper https//arxiv.org/abs/2308.15594☆13Updated 9 months ago
- LieTransformer: Equivariant Self-Attention for Lie Groups☆67Updated 5 months ago
- Code for Canonical 3D Deformable Mapping (C3DM) paper☆19Updated 3 years ago
- Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology☆35Updated 3 years ago
- Julia implementation of SMPL family of 3D human models.☆14Updated last month
- A simple hypernetwork implementation in jax using haiku.☆23Updated 2 years ago
- Implementation of an Attention layer where each head can attend to more than just one token, using coordinate descent to pick topk☆46Updated last year
- Neural Optimal Transport with Lagrangian Costs☆50Updated 6 months ago
- A LinearOperator implementation for PyTorch☆18Updated 4 years ago
- Source-to-Source Debuggable Derivatives in Pure Python☆15Updated 11 months ago
- JAX port of Persistent Independent Particles☆37Updated last year
- Code for "Recognizing Scenes from Novel Viewpoints"☆29Updated 2 years ago
- ☆14Updated 2 years ago
- code associated with paper "Sparse Bayesian Optimization"☆26Updated last year
- ☆25Updated 7 months ago
- Experimental scripts for researching data adaptive learning rate scheduling.☆23Updated last year
- Convolutions and more as einsum for PyTorch☆13Updated 7 months ago
- ☆26Updated 6 months ago