Felix-Petersen / algovision
Differentiable Algorithms and Algorithmic Supervision.
☆111Updated 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☆109Updated last year
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆106Updated 2 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☆125Updated 2 years ago
- Named tensors with first-class dimensions for PyTorch☆322Updated last year
- Stochastic Automatic Differentiation library for PyTorch.☆195Updated 4 months ago
- Use Jax functions in Pytorch☆231Updated last year
- A GPT, made only of MLPs, in Jax☆57Updated 3 years ago
- ☆110Updated last week
- ☆67Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆146Updated last month
- Fast training of unitary deep network layers from low-rank updates☆28Updated 2 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- ☆98Updated 3 years ago
- JMP is a Mixed Precision library for JAX.☆189Updated last month
- OpTree: Optimized PyTree Utilities☆163Updated this week
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 2 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆77Updated 4 years ago
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆67Updated 2 years ago
- ☆100Updated 6 months ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated 8 months ago
- ☆153Updated last year
- Lightweight Cluster/Cloud VM Job Management 🚀☆39Updated 4 months ago
- ☆215Updated 8 months ago
- 🧀 Pytorch code for the Fromage optimiser.☆122Updated 6 months ago
- Meta Optimal Transport☆98Updated last year
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆175Updated 2 years ago
- WandB sweeps integration with Hydra sweeper☆47Updated 11 months ago
- Running Jax in PyTorch Lightning☆83Updated last month