facebookresearch / optimizer-pluginsLinks
A demonstration of source code transformation to implement automatic differentiation, compatible with an operation overload style AD library DiffKt.
☆13Updated 3 years ago
Alternatives and similar repositories for optimizer-plugins
Users that are interested in optimizer-plugins are comparing it to the libraries listed below
Sorting:
- Loop Nest - Linear algebra compiler and code generator.☆22Updated 2 years ago
- Authors implementation of LieTransformer: Equivariant Self-Attention for Lie Groups☆36Updated 4 years ago
- A framework for automatic differentiation in Kotlin☆60Updated 3 years ago
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆17Updated last year
- A LinearOperator implementation for PyTorch☆18Updated 4 years ago
- Texture mapping with variational auto-encoders☆40Updated 3 years ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆43Updated last year
- ☆26Updated last year
- Learning about Neural Radiance Fields☆67Updated 3 years ago
- Implicit^2: Implicit model for implicit neural representations☆29Updated 3 years ago
- LieTransformer: Equivariant Self-Attention for Lie Groups☆72Updated last year
- Implementation of an Attention layer where each head can attend to more than just one token, using coordinate descent to pick topk☆46Updated 2 years ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆27Updated 3 years ago
- ☆81Updated 3 years ago
- Visual Simultaneous Localization and Mapping☆25Updated 3 years ago
- Efficient Householder Transformation in PyTorch☆66Updated 4 years ago
- ☆24Updated 2 months ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆55Updated 2 years ago
- Generic PyTorch implementation of einsum that supports different semirings☆50Updated last year
- ☆89Updated last year
- Experiment Manager☆22Updated 5 years ago
- Webpage☆16Updated last year
- Inference on non-linear dynamical systems written in JAX☆11Updated 5 years ago
- ☆50Updated 4 years ago
- ☆15Updated 4 years ago
- ☆20Updated 2 years ago
- [NeurIPS 2021] Code for Learning Signal-Agnostic Manifolds of Neural Fields☆68Updated 2 years ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆65Updated 10 months ago
- This repository implements and evaluates convolutional networks on the Möbius strip as toy model instantiations of Coordinate Independent…☆72Updated last year
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆87Updated 2 years ago