Felix-Petersen / algovision
Differentiable Algorithms and Algorithmic Supervision.
☆113Updated last year
Alternatives and similar repositories for algovision:
Users that are interested in algovision are comparing it to the libraries listed below
- Differentiable Sorting Networks☆112Updated last year
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆169Updated 3 months ago
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.☆103Updated 3 years ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆257Updated last year
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆126Updated 2 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆108Updated 2 years ago
- Stochastic Automatic Differentiation library for PyTorch.☆198Updated 6 months ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated 10 months ago
- ☆98Updated 3 years ago
- ☆220Updated last month
- ☆112Updated last month
- A GPT, made only of MLPs, in Jax☆57Updated 3 years ago
- Fast training of unitary deep network layers from low-rank updates☆29Updated 2 years ago
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆67Updated 2 years ago
- Lightweight Cluster/Cloud VM Job Management 🚀☆41Updated 6 months ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 3 years ago
- simple bibtex generator for any text with \cite{}☆31Updated 8 months ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆177Updated last month
- ☆49Updated last year
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- A centralized place for deep thinking code and experiments☆82Updated last year
- JMP is a Mixed Precision library for JAX.☆193Updated last month
- Meta Optimal Transport☆100Updated last year
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆185Updated 3 months ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Parameter-Free Optimizers for Pytorch☆122Updated 10 months ago
- Running Jax in PyTorch Lightning☆89Updated 3 months ago
- ☆101Updated 8 months ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆61Updated 3 years ago