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:
- Sparse Boolean linear algebra for Nvidia Cuda, OpenCL and CPU computations☆16Updated 3 years ago
- 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 Kotlin DSL for probabilistic programming.☆12Updated 3 years ago
- Texture mapping with variational auto-encoders☆40Updated 4 years ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆65Updated last year
- LieTransformer: Equivariant Self-Attention for Lie Groups☆77Updated last year
- A framework for automatic differentiation in Kotlin☆60Updated 3 years ago
- Optimal approximate sampling from discrete probability distributions☆18Updated 4 years ago
- Tree Approximate Message Passing☆32Updated last year
- Loop Nest - Linear algebra compiler and code generator.☆21Updated 3 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
- Alphazero on GPU thanks to CUDA.jl☆33Updated 4 years ago
- Library for causal and probabilistic inference in Julia☆12Updated last year
- ☆28Updated 4 years ago
- ☆33Updated 5 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
- ☆109Updated last year
- Efficient Householder Transformation in PyTorch☆69Updated 4 years ago
- ☆30Updated 2 years ago
- Julia implementation of flash-attention operation for neural networks.☆11Updated 2 years ago
- Generic PyTorch implementation of einsum that supports different semirings☆51Updated 2 months ago
- Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch☆97Updated 4 years ago
- ☆26Updated last year
- B-Spline CNNs on Lie groups☆57Updated 6 years ago
- Visual Simultaneous Localization and Mapping☆25Updated 3 years ago
- Differentiable Duckietown☆32Updated 6 years ago
- The differentiable OSQP solver layer for PyTorch☆65Updated 3 years ago
- A LinearOperator implementation for PyTorch☆18Updated 5 years ago