Felix-Petersen / algovision
Differentiable Algorithms and Algorithmic Supervision.
☆115Updated last year
Alternatives and similar repositories for algovision:
Users that are interested in algovision are comparing it to the libraries listed below
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆127Updated 2 years ago
- Stochastic Automatic Differentiation library for PyTorch.☆204Updated 8 months ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆258Updated last year
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆180Updated 3 years ago
- Differentiable Sorting Networks☆114Updated last year
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆109Updated 2 years ago
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.☆105Updated 3 years ago
- Named tensors with first-class dimensions for PyTorch☆320Updated last year
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- ☆67Updated last year
- ☆99Updated 3 years ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆177Updated this week
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated last year
- Use Jax functions in Pytorch☆239Updated last year
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- ☆114Updated this week
- Easy-to-use AdaHessian optimizer (PyTorch)☆78Updated 4 years ago
- JMP is a Mixed Precision library for JAX.☆196Updated 3 months ago
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆173Updated this week
- A GPT, made only of MLPs, in Jax☆57Updated 3 years ago
- Unofficial JAX implementations of deep learning research papers☆156Updated 2 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆124Updated 9 months ago
- ☆156Updated last year
- Running Jax in PyTorch Lightning☆96Updated 4 months ago
- OpTree: Optimized PyTree Utilities☆178Updated this week
- A case study of efficient training of large language models using commodity hardware.☆69Updated 2 years ago
- Meta Optimal Transport☆102Updated last year
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆59Updated 3 years ago
- ☆153Updated 4 years ago