kevinzakka / learn-linalg
Learning some numerical linear algebra.
☆71Updated 4 years ago
Alternatives and similar repositories for learn-linalg:
Users that are interested in learn-linalg are comparing it to the libraries listed below
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆59Updated 4 years ago
- The differentiable OSQP solver layer for PyTorch☆59Updated 2 years ago
- Example python package with pybind11 cpp extension☆57Updated 4 years ago
- Code for NeurIPS 2019 paper: "Symmetry-Based Disentangled Representation Learning requires Interaction with Environments" by H. Caselles-…☆35Updated 5 years ago
- ☆64Updated 7 years ago
- Autoregressive Energy Machines☆77Updated 2 years ago
- ☆28Updated 5 years ago
- The Differentiable Cross-Entropy Method☆126Updated 4 years ago
- A differentiation API for PyTorch☆30Updated 4 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Practice with MCMC methods and dynamics (Langevin, Hamiltonian, etc.)☆42Updated 5 years ago
- Normalizing Flows in Jax☆107Updated 4 years ago
- Performant, differentiable reinforcement learning☆120Updated 10 months ago
- ☆45Updated 5 years ago
- Material for my Caltech tutorial on deep learning and tensor methods☆70Updated 6 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago
- Repo for the Tutorials of Day1-Day3 of the Nordic Probabilistic AI School 2019 (https://2019.probabilistic.ai/)☆20Updated 5 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- A collection of LaTeX shortcuts for commonly used mathematical expressions in machine learning.☆58Updated 5 years ago
- A lightweight library for tensorflow 2.0☆66Updated 5 years ago
- Code for Unbiased Implicit Variational Inference (UIVI)☆14Updated 6 years ago
- ☆85Updated 4 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 5 years ago
- ☆67Updated last year
- We simulate a wind tunnel, place a rectangular occlusion in it, and then use gradient descent to turn the occlusion into a wing.☆25Updated 4 years ago
- ☆29Updated 5 years ago
- Graduate topics course on learning discrete latent structure.☆67Updated 6 years ago
- Understanding nuts and bolts of neural networks with PyTorch☆33Updated 4 years ago
- ☆30Updated 4 years ago