Alternatives and similar repositories for torchjd

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

• iShohei220 / adopt
  Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"
  ☆434Updated last year
• nikhilvyas / SOAP
  ☆246Updated last year
• kozistr / pytorch_optimizer
  optimizer & lr scheduler & loss function collections in PyTorch
  ☆388Updated last week
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆352Updated 2 months ago
• 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
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆186Updated last year
• galilai-group / stable-pretraining
  Reliable, minimal and scalable library for pretraining foundation and world models
  ☆126Updated this week
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆281Updated last month
• kach / gradient-descent-the-ultimate-optimizer
  Code for our NeurIPS 2022 paper
  ☆371Updated 3 years ago
• kvfrans / splus
  ☆123Updated 7 months ago
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆130Updated last year
• j3soon / tbparse
  Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow
  ☆208Updated last year
• Algomancer / Bayesian-Flow-Networks
  A simple implimentation of Bayesian Flow Networks (BFN)
  ☆240Updated 2 years ago
• locuslab / torchdeq
  Modern Fixed Point Systems using Pytorch
  ☆127Updated 2 years ago
• fferflo / einx
  Universal Notation for Tensor Operations in Python.
  ☆469Updated 10 months ago
• BlackHC / neural_net_checklist
  ☆215Updated last year
• facebookresearch / dadaptation
  D-Adaptation for SGD, Adam and AdaGrad
  ☆528Updated last year
• cooper-org / cooper
  A general-purpose, deep learning-first library for constrained optimization in PyTorch
  ☆152Updated 2 months ago
• cloneofsimo / d3pm
  Minimal Implementation of a D3PM in pytorch
  ☆285Updated last year
• probabilists / zuko
  Normalizing flows in PyTorch
  ☆442Updated 2 months ago
• shyamsn97 / hyper-nn
  Easy Hypernetworks in Pytorch and Jax
  ☆106Updated 3 years ago
• lindermanlab / S5
  ☆314Updated last year
• MinghuiChen43 / awesome-deep-phenomena
  A curated list of papers of interesting empirical study and insight on deep learning. Continually updating...
  ☆390Updated last month
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆190Updated 3 weeks ago
• stanislavfort / dissect-git-re-basin
  Replicating and dissecting the git-re-basin project in one-click-replication Colabs
  ☆37Updated 3 years ago
• nnaisense / bayesian-flow-networks
  This is the official code release for Bayesian Flow Networks.
  ☆309Updated last year
• lucidrains / gradnorm-pytorch
  A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch
  ☆126Updated 5 months ago
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆195Updated 10 months ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆558Updated 3 weeks ago
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆109Updated 3 months ago