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:
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆118Updated 3 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 3 years ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆259Updated 2 years ago
- Differentiable Sorting Networks☆125Updated 2 years ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆63Updated 4 years ago
- Stochastic Automatic Differentiation library for PyTorch.☆208Updated last year
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆131Updated 3 years ago
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.☆111Updated 4 years ago
- Fast Discounted Cumulative Sums in PyTorch☆96Updated 4 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆45Updated 5 months ago
- ☆71Updated 2 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆39Updated 3 years ago
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆67Updated 3 years ago
- ☆100Updated 4 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆79Updated 5 years ago
- Image augmentation library for Jax☆41Updated last year
- Meta Optimal Transport☆106Updated 2 years ago
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆129Updated 2 years ago
- Implementation of deep implicit attention in PyTorch☆65Updated 4 years ago
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorch☆25Updated 5 years ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆184Updated 4 years ago
- a lightweight transformer library for PyTorch☆72Updated 4 years ago
- Efficient Householder Transformation in PyTorch☆69Updated 4 years ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆181Updated 8 months ago
- A custom PyTorch layer that is capable of implementing extremely wide and sparse linear layers efficiently☆51Updated 2 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆188Updated 2 weeks ago
- DeepOBS: A Deep Learning Optimizer Benchmark Suite☆109Updated 2 years ago
- Open source code for EigenGame.☆34Updated 2 years ago
- Lightweight Cluster/Cloud VM Job Management 🚀☆42Updated last year
- Convert scikit-learn models to PyTorch modules☆168Updated last year