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☆58Updated 4 years ago
- Example python package with pybind11 cpp extension☆57Updated 4 years ago
- The differentiable OSQP solver layer for PyTorch☆58Updated 2 years ago
- Tensorflow 2.0 implementation of Fourier Feature Mapping Networks.☆42Updated 4 years ago
- ☆85Updated 4 years ago
- PyTorch implementation comparison of old and new method of determining eigenvectors from eigenvalues.☆99Updated 3 years ago
- ☆63Updated 6 years ago
- Riemannian Convex Potential Maps☆67Updated last year
- Source code for ICLR 2020 paper: "Learning to Guide Random Search"☆39Updated 4 months ago
- Progress, Notes, Summaries and a lot of Questions on Machine Learning☆55Updated 5 years ago
- Performant, differentiable reinforcement learning☆120Updated 7 months ago
- Some small scale experiments for my blog posts 📝☆78Updated 2 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆43Updated 3 years ago
- ☆29Updated 4 years ago
- Normalizing Flows in Jax☆106Updated 4 years ago
- Material for my Caltech tutorial on deep learning and tensor methods☆70Updated 6 years ago
- Interpolation routines in Pytorch.☆110Updated 4 years ago
- Practice with MCMC methods and dynamics (Langevin, Hamiltonian, etc.)☆42Updated 4 years ago
- 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
- The Differentiable Cross-Entropy Method☆125Updated 4 years ago
- Understanding ML and deep learning through geometry☆158Updated 3 years ago
- Experiment code for "Randomized Automatic Differentiation"☆66Updated 4 years ago
- ☆28Updated 5 years ago
- Very deep VAEs in JAX/Flax☆46Updated 3 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Code for NeurIPS 2019 paper: "Symmetry-Based Disentangled Representation Learning requires Interaction with Environments" by H. Caselles-…☆35Updated 5 years ago
- Repo for the Tutorials of Day1-Day3 of the Nordic Probabilistic AI School 2019 (https://2019.probabilistic.ai/)☆20Updated 5 years ago
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆98Updated last year
- ☆45Updated 5 years ago
- A differentiation API for PyTorch☆30Updated 4 years ago