formll / dog

Related projects ⓘ

Alternatives and complementary repositories for dog

• stanislavfort / dissect-git-re-basin
  Replicating and dissecting the git-re-basin project in one-click-replication Colabs
  ☆36Updated 2 years ago
• nikhilvyas / SOAP
  ☆128Updated this week
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆109Updated 6 months ago
• shikaiqiu / compute-better-spent
  ☆46Updated last month
• epfml / llm-baselines
  NanoGPT-like codebase for LLM training
  ☆75Updated this week
• locuslab / edge-of-stability
  ☆59Updated 3 years ago
• zyushun / hessian-spectrum
  Code for the paper: Why Transformers Need Adam: A Hessian Perspective
  ☆42Updated 6 months ago
• aks2203 / deep-thinking
  A centralized place for deep thinking code and experiments
  ☆77Updated last year
• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆63Updated 2 years ago
• Ping-C / optimizer
  This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…
  ☆34Updated last year
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆95Updated 6 months ago
• gortizji / linearized-networks
  Source code of "What can linearized neural networks actually say about generalization?
  ☆18Updated 3 years ago
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆40Updated 2 years ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆79Updated 9 months ago
• optimizedlearning / mechanic
  ☆35Updated 9 months ago
• jxbz / modula
  Scalable neural net training via automatic normalization in the modular norm.
  ☆121Updated 3 months ago
• JeanKaddour / NoTrainNoGain
  Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)
  ☆79Updated last year
• edwardjhu / TP4
  Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)
  ☆58Updated 3 years ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …
  ☆127Updated last month
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆132Updated 8 months ago
• Leiay / looped_transformer
  ☆20Updated 11 months ago
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆45Updated last year
• cloneofsimo / zeroshampoo
  ☆31Updated 2 months ago
• KindXiaoming / Omnigrok
  Omnigrok: Grokking Beyond Algorithmic Data
  ☆49Updated last year
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆70Updated 6 months ago
• google-deepmind / conformal_training
  This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…
  ☆121Updated 2 years ago
• google-research / jax-influence
  ☆58Updated 2 years ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders in JAX.
  ☆59Updated last month
• runame / laplace-redux
  Laplace Redux -- Effortless Bayesian Deep Learning
  ☆38Updated last year
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆44Updated last year