TorchJD / torchjdLinks
Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning).
☆287Updated this week
Alternatives and similar repositories for torchjd
Users that are interested in torchjd are comparing it to the libraries listed below
Sorting:
- ☆227Updated last year
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆122Updated 9 months ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆330Updated 3 weeks ago
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆429Updated last year
- optimizer & lr scheduler & loss function collections in PyTorch☆376Updated last week
- The AdEMAMix Optimizer: Better, Faster, Older.☆186Updated last year
- Reliable, minimal and scalable library for pretraining foundation and world models☆107Updated 3 weeks ago
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Efficient optimizers☆275Updated last month
- Code for our NeurIPS 2022 paper☆370Updated 2 years ago
- ☆121Updated 6 months ago
- A simple implimentation of Bayesian Flow Networks (BFN)☆240Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆188Updated last week
- Modern Fixed Point Systems using Pytorch☆125Updated 2 years ago
- ☆211Updated last year
- Normalizing flows in PyTorch☆428Updated last week
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆146Updated 3 weeks ago
- TorchOpt is an efficient library for differentiable optimization built upon PyTorch.☆620Updated last week
- For optimization algorithm research and development.☆548Updated 3 weeks ago
- ☆312Updated 11 months ago
- A curated list of papers of interesting empirical study and insight on deep learning. Continually updating...☆382Updated last week
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆36Updated 3 years ago
- Laplace approximations for Deep Learning.☆529Updated 7 months ago
- D-Adaptation for SGD, Adam and AdaGrad☆528Updated 10 months ago
- Easy Hypernetworks in Pytorch and Jax☆106Updated 2 years ago
- Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow☆206Updated last year
- Universal Notation for Tensor Operations in Python.☆452Updated 8 months ago
- Implementation of the proposed minGRU in Pytorch☆311Updated 9 months ago
- Getting crystal-like representations with harmonic loss☆192Updated 8 months ago
- Simple (and cheap!) neural network uncertainty estimation☆78Updated 2 months ago