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:
- A framework for automatic differentiation in Kotlin☆61Updated 3 years ago
- Sparse Boolean linear algebra for Nvidia Cuda, OpenCL and CPU computations☆15Updated 3 years ago
- Authors implementation of LieTransformer: Equivariant Self-Attention for Lie Groups☆36Updated 4 years ago
- Texture mapping with variational auto-encoders☆40Updated 4 years ago
- A LinearOperator implementation for PyTorch☆18Updated 4 years ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆44Updated 2 years ago
- ☆33Updated 4 years ago
- Efficient Householder Transformation in PyTorch☆66Updated 4 years ago
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆18Updated last year
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆88Updated 3 years ago
- Loop Nest - Linear algebra compiler and code generator.☆21Updated 3 years ago
- ☆107Updated last year
- ☆15Updated 3 years ago
- Neural Fixed-Point Acceleration for Convex Optimization☆29Updated 3 years ago
- ☆81Updated 3 years ago
- code associated with paper "Sparse Bayesian Optimization"☆26Updated 2 years ago
- Convolutions and more as einsum for PyTorch☆17Updated last year
- A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes☆25Updated 4 years ago
- Pytorch implementation of SuperPolyak subgradient method.☆43Updated 3 years ago
- B-Spline CNNs on Lie groups☆56Updated 5 years ago
- ☆15Updated 4 years ago
- Utility functions that I reuse across different projects☆14Updated 4 years ago
- Implementation of Bayesian Sum-Product Networks☆13Updated 5 years ago
- ☆26Updated last year
- Pytrees + dataclasses ❤️☆67Updated 6 months ago
- Inference on non-linear dynamical systems written in JAX☆11Updated 5 years ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆28Updated 4 years ago
- The differentiable OSQP solver layer for PyTorch☆64Updated 2 years ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆65Updated last year
- Generic PyTorch implementation of einsum that supports different semirings☆51Updated last year