Felix-Petersen / algovisionLinks
Differentiable Algorithms and Algorithmic Supervision.
☆116Updated 2 years ago
Alternatives and similar repositories for algovision
Users that are interested in algovision are comparing it to the libraries listed below
Sorting:
- Differentiable Sorting Networks☆117Updated 2 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆112Updated 3 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆128Updated 3 years ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆258Updated last year
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Stochastic Automatic Differentiation library for PyTorch.☆205Updated 11 months ago
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.☆105Updated 4 years ago
- Meta Optimal Transport☆103Updated 2 years ago
- Convert scikit-learn models to PyTorch modules☆163Updated last year
- Easy-to-use AdaHessian optimizer (PyTorch)☆79Updated 4 years ago
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆59Updated 3 years ago
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆67Updated 3 years ago
- Lightweight Cluster/Cloud VM Job Management 🚀☆42Updated 11 months ago
- 🧀 Pytorch code for the Fromage optimiser.☆125Updated last year
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 3 weeks ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆62Updated 4 years ago
- Unofficial JAX implementations of deep learning research papers☆156Updated 3 years ago
- Named tensors with first-class dimensions for PyTorch☆332Updated 2 years ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 4 years ago
- ☆100Updated 3 years ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆181Updated 4 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆179Updated last week
- A centralized place for deep thinking code and experiments☆85Updated 2 years ago
- Implementation of Hierarchical Transformer Memory (HTM) for Pytorch☆75Updated 3 years ago
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆81Updated 3 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated last year
- ☆68Updated last year
- Fast Discounted Cumulative Sums in PyTorch☆96Updated 3 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago