Alternatives and similar repositories for torchjd

Users that are interested in torchjd are comparing it to the libraries listed below

• nikhilvyas / SOAP
  ☆238Updated last year
• iShohei220 / adopt
  Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"
  ☆434Updated last year
• kozistr / pytorch_optimizer
  optimizer & lr scheduler & loss function collections in PyTorch
  ☆388Updated last week
• epistoteles / TensorHue
  TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…
  ☆124Updated 11 months ago
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆280Updated last month
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆347Updated 2 months ago
• kach / gradient-descent-the-ultimate-optimizer
  Code for our NeurIPS 2022 paper
  ☆371Updated 3 years ago
• locuslab / torchdeq
  Modern Fixed Point Systems using Pytorch
  ☆125Updated 2 years ago
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆186Updated last year
• j3soon / tbparse
  Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow
  ☆208Updated last year
• facebookresearch / dadaptation
  D-Adaptation for SGD, Adam and AdaGrad
  ☆528Updated last year
• lindermanlab / S5
  ☆314Updated last year
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆191Updated 2 weeks ago
• lucidrains / minGRU-pytorch
  Implementation of the proposed minGRU in Pytorch
  ☆318Updated last month
• galilai-group / stable-pretraining
  Reliable, minimal and scalable library for pretraining foundation and world models
  ☆118Updated last week
• kvfrans / splus
  ☆123Updated 7 months ago
• BlackHC / neural_net_checklist
  ☆215Updated last year
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆130Updated last year
• metaopt / torchopt
  TorchOpt is an efficient library for differentiable optimization built upon PyTorch.
  ☆623Updated 3 weeks ago
• cooper-org / cooper
  A general-purpose, deep learning-first library for constrained optimization in PyTorch
  ☆151Updated 2 months ago
• Algomancer / Bayesian-Flow-Networks
  A simple implimentation of Bayesian Flow Networks (BFN)
  ☆240Updated 2 years ago
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆107Updated 3 months ago
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆126Updated last year
• aleximmer / Laplace
  Laplace approximations for Deep Learning.
  ☆532Updated 9 months ago
• cloneofsimo / d3pm
  Minimal Implementation of a D3PM in pytorch
  ☆285Updated last year
• fferflo / einx
  Universal Notation for Tensor Operations in Python.
  ☆463Updated 9 months ago
• mlcommons / algorithmic-efficiency
  MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…
  ☆406Updated last week
• shyamsn97 / hyper-nn
  Easy Hypernetworks in Pytorch and Jax
  ☆106Updated 3 years ago
• lucidrains / gradnorm-pytorch
  A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch
  ☆126Updated 5 months ago
• kyleliang919 / C-Optim
  When it comes to optimizers, it's always better to be safe than sorry
  ☆399Updated 4 months ago