KellerJordan / top-sgd

Related projects: ⓘ

• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated last year
• shikaiqiu / compute-better-spent
  ☆42Updated 3 months ago
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆29Updated 2 months ago
• Artur-Galstyan / statedict2pytree
  ☆40Updated 2 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆74Updated 7 months ago
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆43Updated last year
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆28Updated 4 months ago
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆57Updated last month
• stanislavfort / dissect-git-re-basin
  Replicating and dissecting the git-re-basin project in one-click-replication Colabs
  ☆36Updated 2 years ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆45Updated last month
• andrew-silva / clean-rl-mlx
  Clean RL implementation using MLX
  ☆26Updated 6 months ago
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆37Updated 2 months ago
• radarFudan / mamba-minimal-jax
  ☆28Updated last week
• google-deepmind / eigengame
  Open source code for EigenGame.
  ☆28Updated last year
• mgrankin / minGPT
  minGPT in JAX
  ☆45Updated 2 years ago
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆20Updated 2 months ago
• aks2203 / easy-to-hard
  Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"
  ☆58Updated 2 years ago
• AllanYangZhou / universal_neural_functional
  ☆46Updated 7 months ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …
  ☆105Updated this week
• 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…
  ☆22Updated 10 months ago
• srush / Tensor-Puzzles-Penzai
  ☆17Updated 4 months ago
• vvvm23 / mezo-jax
  JAX implementation of "Fine-Tuning Language Models with Just Forward Passes"
  ☆20Updated last year
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆66Updated 7 months ago
• smonsays / metax
  flexible meta-learning in jax
  ☆12Updated 11 months ago
• lucidrains / hyena-pytorch
  ☆23Updated this week
• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆36Updated last year
• andyehrenberg / flaxlm
  ☆27Updated this week
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated last month
• nikhilvyas / SOAP
  ☆30Updated this week
• NVlabs / Forecasting-Model-Search
  A system for automating selection and optimization of pre-trained models from the TAO Model Zoo
  ☆19Updated 2 months ago