Alternatives and similar repositories for dog:

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

• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆122Updated 11 months ago
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆45Updated 2 years ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆38Updated 3 months 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…
  ☆35Updated 2 years ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆83Updated last year
• shikaiqiu / compute-better-spent
  ☆52Updated 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
• locuslab / edge-of-stability
  ☆65Updated 3 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆173Updated 2 weeks ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆135Updated last year
• f-dangel / curvlinops
  PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)
  ☆34Updated 3 weeks ago
• nikhilvyas / SOAP
  ☆169Updated 3 months ago
• aradha / recursive_feature_machines
  ☆43Updated this week
• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆87Updated 9 months ago
• edwardjhu / TP4
  Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)
  ☆61Updated 3 years ago
• KindXiaoming / Omnigrok
  Omnigrok: Grokking Beyond Algorithmic Data
  ☆53Updated 2 years ago
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆48Updated last year
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆170Updated 3 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆86Updated 7 months ago
• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆76Updated 2 years ago
• tml-epfl / why-weight-decay
  Why Do We Need Weight Decay in Modern Deep Learning? [NeurIPS 2024]
  ☆63Updated 6 months ago
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆46Updated last year
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆77Updated 5 months ago
• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆56Updated last year
• f-dangel / singd
  [ICML 2024] SINGD: KFAC-like Structured Inverse-Free Natural Gradient Descent (http://arxiv.org/abs/2312.05705)
  ☆21Updated 4 months ago
• danielmamay / grokking
  Implementation of OpenAI's 'Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets' paper.
  ☆35Updated last year
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆108Updated 3 months ago
• google-deepmind / md4
  Official Jax Implementation of MD4 Masked Diffusion Models
  ☆69Updated 3 weeks ago
• cgarciae / simple_flow_matching
  ☆18Updated 3 months ago
• VectorInstitute / vbll
  Simple (and cheap!) neural network uncertainty estimation
  ☆61Updated last month