Alternatives and similar repositories for top-sgd:

Users that are interested in top-sgd are comparing it to the libraries listed below

• stanislavfort / dissect-git-re-basin
  Replicating and dissecting the git-re-basin project in one-click-replication Colabs
  ☆36Updated 2 years ago
• shikaiqiu / compute-better-spent
  ☆50Updated 3 months ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆148Updated last month
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆44Updated last year
• apple / ml-ademamix
  ☆52Updated 2 months ago
• wiseodd / laplace-bayesopt
  Laplace approximated BNN surrogate for BoTorch
  ☆8Updated 2 months ago
• facebookresearch / meta-ot
  Meta Optimal Transport
  ☆98Updated last year
• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆22Updated 2 weeks ago
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆34Updated last year
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆82Updated 11 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆36Updated last month
• gisilvs / AEF
  ☆33Updated last year
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆44Updated 2 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆106Updated 2 years ago
• stanfordmlgroup / disentanglement
  Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology
  ☆35Updated 3 years ago
• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆87Updated 7 months ago
• google-deepmind / eigengame
  Open source code for EigenGame.
  ☆29Updated last year
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆44Updated last year
• Rose-STL-Lab / AutoSTPP
  Automatic Integration for Neural Spatio-Temporal Point Process models (AI-STPP) is a new paradigm for exact, efficient, non-parametric inf…
  ☆24Updated 3 months ago
• vvvm23 / mezo-jax
  JAX implementation of "Fine-Tuning Language Models with Just Forward Passes"
  ☆19Updated last year
• dasayan05 / iclr24_blog_code
  Code repo for ICLR 24 BlogPost titled "Building Diffusion Model's theory from ground up"
  ☆15Updated last year
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆75Updated last month
• facebookresearch / projUNN
  Fast training of unitary deep network layers from low-rank updates
  ☆28Updated 2 years ago
• johnryan465 / pscan
  ☆37Updated last year
• AllanYangZhou / universal_neural_functional
  ☆48Updated 11 months ago
• NVlabs / Forecasting-Model-Search
  A system for automating selection and optimization of pre-trained models from the TAO Model Zoo
  ☆24Updated 6 months ago
• zacharyhorvitz / Fk-Diffusion-Steering
  A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.
  ☆40Updated this week
• Sanaelotfi / Bayesian_model_comparison
  Supporing code for the paper "Bayesian Model Selection, the Marginal Likelihood, and Generalization".
  ☆35Updated 2 years ago
• TorchDR / TorchDR
  TorchDR - PyTorch Dimensionality Reduction
  ☆73Updated this week
• dvruette / barrel-rec-pytorch
  ☆53Updated last year