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:
- Authors implementation of LieTransformer: Equivariant Self-Attention for Lie Groups☆36Updated 5 years ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆45Updated 2 years ago
- A framework for automatic differentiation in Kotlin☆60Updated 3 years ago
- LieTransformer: Equivariant Self-Attention for Lie Groups☆77Updated last year
- Loop Nest - Linear algebra compiler and code generator.☆21Updated 3 years ago
- Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch☆97Updated 4 years ago
- ☆34Updated 2 years ago
- Texture mapping with variational auto-encoders☆40Updated 4 years ago
- Experiments for the AISTATS publication on Reparameterizing Distributions over Lie Groups☆52Updated 3 years ago
- A differentiation API for PyTorch☆30Updated 5 years ago
- ☆15Updated 5 years ago
- To be a next-generation DL-based phenotype prediction from genome mutations.☆19Updated 4 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆89Updated 3 years ago
- This repository implements and evaluates convolutional networks on the Möbius strip as toy model instantiations of Coordinate Independent…☆76Updated 2 years ago
- Generic PyTorch implementation of einsum that supports different semirings☆51Updated 2 months ago
- Sparse Boolean linear algebra for Nvidia Cuda, OpenCL and CPU computations☆16Updated 3 years ago
- Efficient Householder Transformation in PyTorch☆69Updated 4 years ago
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆67Updated 3 years ago
- ☆33Updated 5 years ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆65Updated last year
- ☆109Updated last year
- Tree Approximate Message Passing☆32Updated last year
- A LinearOperator implementation for PyTorch☆18Updated 5 years ago
- Julia implementation of flash-attention operation for neural networks.☆11Updated 2 years ago
- Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"☆59Updated 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☆47Updated 2 years ago
- ☆45Updated this week
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- JAX exponential map normalising flows on sphere☆17Updated 5 years ago
- pytorch implementation of trDesign☆45Updated 4 years ago