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:
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆45Updated 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
- Sparse Boolean linear algebra for Nvidia Cuda, OpenCL and CPU computations☆16Updated 3 years ago
- LieTransformer: Equivariant Self-Attention for Lie Groups☆77Updated last year
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆65Updated last year
- Texture mapping with variational auto-encoders☆40Updated 4 years ago
- Julia implementation of flash-attention operation for neural networks.☆11Updated 2 years ago
- Experiments for the AISTATS publication on Reparameterizing Distributions over Lie Groups☆52Updated 3 years ago
- ☆107Updated last year
- ☆15Updated 5 years ago
- Tree Approximate Message Passing☆32Updated last year
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆18Updated last year
- Efficient Householder Transformation in PyTorch☆69Updated 4 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆89Updated 3 years ago
- A LinearOperator implementation for PyTorch☆18Updated 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
- Meta-learning inductive biases in the form of useful conserved quantities.☆39Updated 3 years ago
- ☆33Updated 5 years ago
- Library for causal and probabilistic inference in Julia☆12Updated 11 months ago
- Visual Simultaneous Localization and Mapping☆25Updated 3 years ago
- Generic PyTorch implementation of einsum that supports different semirings☆51Updated last month
- Alphazero on GPU thanks to CUDA.jl☆33Updated 4 years ago
- Implicit^2: Implicit model for implicit neural representations☆29Updated 4 years ago
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆67Updated 3 years ago
- Pytrees + dataclasses ❤️☆71Updated last month
- MDP and RL interface for PDDL domains via PDDL.jl + POMDPs.jl.☆16Updated last year
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆42Updated 5 years ago
- ☆28Updated 4 years ago