linclab / EG_optimiser

Related projects ⓘ

Alternatives and complementary repositories for EG_optimiser

• DAIOS-AI / mindscript
  A programming language for formal/informal computation.
  ☆41Updated 5 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…
  ☆34Updated last year
• joschu / jax-exp
  ☆24Updated 5 years ago
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆30Updated 4 months ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …
  ☆128Updated last month
• Artur-Galstyan / statedict2pytree
  ☆40Updated 4 months ago
• GallagherCommaJack / modulax
  ☆16Updated 3 months ago
• srush / torch-golf
  Silly twitter torch implementations.
  ☆46Updated 2 years ago
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆19Updated 2 weeks ago
• cgarciae / einop
  ☆58Updated 2 years ago
• davisyoshida / qax
  If it quacks like a tensor...
  ☆52Updated last week
• AllanYangZhou / universal_neural_functional
  ☆48Updated 9 months ago
• vvvm23 / mezo-jax
  JAX implementation of "Fine-Tuning Language Models with Just Forward Passes"
  ☆20Updated last year
• google-deepmind / einshape
  ☆99Updated 4 months ago
• probcomp / hfppl
  Probabilistic programming with HuggingFace language models
  ☆89Updated this week
• yashbonde / rasp
  Implementing RASP transformer programming language https://arxiv.org/pdf/2106.06981.pdf.
  ☆44Updated 3 years ago
• akandykeller / NeuralWaveMachines
  Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …
  ☆69Updated last year
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆127Updated 3 weeks ago
• samacqua / LARC
  Language-annotated Abstraction and Reasoning Corpus
  ☆79Updated last year
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆38Updated 4 years ago
• dlwh / jax_sourceror
  Turn jitted jax functions back into python source code
  ☆20Updated 4 months ago
• harvardnlp / strux
  ☆18Updated 2 years ago
• ethancaballero / broken_neural_scaling_laws
  Code Release for "Broken Neural Scaling Laws" (BNSL) paper
  ☆57Updated last year
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆152Updated this week
• j-towns / vdvae-jax
  Very deep VAEs in JAX/Flax
  ☆45Updated 3 years ago
• modula-systems / modula
  Scalable neural net training via automatic normalization in the modular norm.
  ☆122Updated this week
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆43Updated last year
• srush / learns-dex
  ☆33Updated 3 years ago
• dvruette / barrel-rec-pytorch
  ☆53Updated 10 months ago