TorchJD / torchjd
Library for Jacobian descent with PyTorch. It enables optimization of neural networks with multiple losses (e.g. multi-task learning).
☆220Updated this week
Alternatives and similar repositories for torchjd:
Users that are interested in torchjd are comparing it to the libraries listed below
- ☆171Updated 3 months ago
- Efficient optimizers☆185Updated this week
- The AdEMAMix Optimizer: Better, Faster, Older.☆179Updated 6 months ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆222Updated 3 weeks ago
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆420Updated 3 months ago
- 🧱 Modula software package☆176Updated this week
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆114Updated last month
- ☆286Updated 2 months ago
- When it comes to optimizers, it's always better to be safe than sorry☆214Updated this week
- Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.…☆90Updated 2 months ago
- D-Adaptation for SGD, Adam and AdaGrad☆519Updated 2 months ago
- Implementation of the proposed minGRU in Pytorch☆283Updated 2 weeks ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆120Updated 8 months ago
- ☆150Updated 7 months ago
- Accelerated First Order Parallel Associative Scan☆177Updated 7 months ago
- For optimization algorithm research and development.☆503Updated this week
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆361Updated last month
- A State-Space Model with Rational Transfer Function Representation.☆78Updated 10 months ago
- Scalable and Performant Data Loading☆230Updated this week
- Code for our NeurIPS 2022 paper☆365Updated 2 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆171Updated this week
- A repository for log-time feedforward networks☆220Updated 11 months ago
- ☆36Updated last year
- optimizer & lr scheduler & loss function collections in PyTorch☆280Updated this week
- ☆145Updated last year
- Parameter-Free Optimizers for Pytorch☆122Updated 11 months ago
- A curated list of papers of interesting empirical study and insight on deep learning. Continually updating...☆314Updated last week
- FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores☆307Updated 3 months ago
- ☆164Updated 2 years ago
- Laplace approximations for Deep Learning.☆498Updated last month