vvvm23 / mezo-jax

Related projects ⓘ

Alternatives and complementary repositories for mezo-jax

• shikaiqiu / compute-better-spent
  ☆46Updated last month
• 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
• smonsays / jax-hypernetwork
  A simple hypernetwork implementation in jax using haiku.
  ☆23Updated 2 years ago
• aks2203 / easy-to-hard
  Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"
  ☆60Updated 2 years ago
• AndyShih12 / LongHorizonTemperatureScaling
  PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023
  ☆19Updated last year
• AllanYangZhou / universal_neural_functional
  ☆48Updated 9 months ago
• yilundu / irem_code_release
  ICML 2022: Learning Iterative Reasoning through Energy Minimization
  ☆44Updated last year
• IBM / ColPret
  Efficient Scaling laws and collaborative pretraining.
  ☆13Updated last week
• yilundu / ebm_compositionality
  [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models
  ☆43Updated last year
• cgarciae / einop
  ☆58Updated 2 years ago
• gregorbachmann / scaling_mlps
  ☆51Updated 5 months ago
• Artur-Galstyan / statedict2pytree
  ☆40Updated 4 months ago
• facebookresearch / projUNN
  Fast training of unitary deep network layers from low-rank updates
  ☆28Updated last year
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆79Updated 9 months ago
• AndPotap / einsum-search
  ☆25Updated last month
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• 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 last month
• ethz-spylab / superhuman-ai-consistency
  ☆28Updated last year
• cloneofsimo / zeroshampoo
  ☆31Updated 2 months ago
• joschu / jax-exp
  ☆24Updated 5 years ago
• smonsays / hypernetwork-attention
  Official code for the paper "Attention as a Hypernetwork"
  ☆23Updated 5 months ago
• JonasGeiping / dataaugs
  ☆17Updated 2 years ago
• shoaibahmed / metadata_archaeology
  Official code for the paper: "Metadata Archaeology"
  ☆18Updated last year
• aks2203 / deep-thinking
  A centralized place for deep thinking code and experiments
  ☆77Updated last year
• ermongroup / fast_feedforward_computation
  Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021
  ☆26Updated 3 years ago
• katiekang1998 / reasoning_generalization
  ☆16Updated last month
• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆32Updated 3 years ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆72Updated 10 months ago
• EleutherAI / rnngineering
  Engineering the state of RNN language models (Mamba, RWKV, etc.)
  ☆32Updated 5 months ago