Alternatives and similar repositories for torchjd

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

• nikhilvyas / SOAP
  ☆190Updated 6 months ago
• HomebrewML / HeavyBall
  Efficient optimizers
  ☆220Updated 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…
  ☆117Updated 4 months ago
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆252Updated 3 months ago
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆183Updated 9 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆200Updated 3 months ago
• locuslab / torchdeq
  Modern Fixed Point Systems using Pytorch
  ☆94Updated last year
• BlackHC / neural_net_checklist
  ☆150Updated 10 months ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆177Updated 2 weeks ago
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆130Updated last year
• lindermanlab / S5
  ☆292Updated 5 months ago
• stanislavfort / dissect-git-re-basin
  Replicating and dissecting the git-re-basin project in one-click-replication Colabs
  ☆36Updated 2 years ago
• iShohei220 / adopt
  Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"
  ☆426Updated 6 months ago
• optimizedlearning / mechanic
  ☆36Updated last year
• logancyang / loss-landscape-anim
  Create animations for the optimization trajectory of neural nets
  ☆156Updated last year
• HazyResearch / flash-fft-conv
  FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores
  ☆319Updated 6 months ago
• formll / dog
  DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule
  ☆63Updated last year
• lucidrains / gradnorm-pytorch
  A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch
  ☆97Updated last year
• pbelcak / fastfeedforward
  A repository for log-time feedforward networks
  ☆222Updated last year
• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆91Updated 2 months ago
• cooper-org / cooper
  A general-purpose, deep learning-first library for constrained optimization in PyTorch
  ☆125Updated 2 weeks ago
• ctlllll / SGConv
  ☆163Updated 2 years ago
• kach / gradient-descent-the-ultimate-optimizer
  Code for our NeurIPS 2022 paper
  ☆368Updated 2 years ago
• pytorch / torcheval
  A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…
  ☆233Updated 5 months ago
• f-dangel / curvlinops
  PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)
  ☆40Updated 2 months ago
• probabilists / zuko
  Normalizing flows in PyTorch
  ☆391Updated 3 weeks ago
• VectorInstitute / vbll
  Simple (and cheap!) neural network uncertainty estimation
  ☆66Updated last month
• kvfrans / splus
  ☆104Updated 2 weeks ago
• NX-AI / xlstm-jax
  Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.…
  ☆94Updated 5 months ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆156Updated 3 years ago