AllanYangZhou / universal_neural_functional

Related projects ⓘ

Alternatives and complementary repositories for universal_neural_functional

• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆53Updated 11 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
• shikaiqiu / compute-better-spent
  ☆46Updated last month
• Artur-Galstyan / statedict2pytree
  ☆40Updated 4 months ago
• AllanYangZhou / nfn
  NF-Layers for constructing neural functionals.
  ☆75Updated 10 months ago
• yilundu / irem_code_release
  ICML 2022: Learning Iterative Reasoning through Energy Minimization
  ☆43Updated last year
• radarFudan / mamba-minimal-jax
  ☆29Updated 2 months ago
• mkofinas / neural-graphs
  Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).
  ☆70Updated 3 months ago
• jxbz / modula
  Scalable neural net training via automatic normalization in the modular norm.
  ☆121Updated 3 months ago
• enajx / HyperNCA
  ☆36Updated 2 years ago
• dvruette / barrel-rec-pytorch
  ☆53Updated 10 months ago
• yilundu / ired_code_release
  ☆46Updated 5 months ago
• xu3kev / BARC
  Bootstrapping ARC
  ☆63Updated this week
• davisyoshida / qax
  If it quacks like a tensor...
  ☆52Updated last week
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆77Updated 3 months ago
• AhmedImtiazPrio / grok-adversarial
  Deep Networks Grok All the Time and Here is Why
  ☆17Updated 6 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆79Updated 9 months ago
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆49Updated 4 months ago
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆21Updated 4 months ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆26Updated 2 weeks ago
• SamsungSAILMontreal / ghn3
  Code for "Can We Scale Transformers to Predict Parameters of Diverse ImageNet Models?" [ICML 2023]
  ☆31Updated 2 months ago
• Silent-Zebra / twisted-smc-lm
  ☆22Updated this week
• oripress / EntropyEnigma
  Official code for the ICML 2024 paper "The Entropy Enigma: Success and Failure of Entropy Minimization"
  ☆44Updated 5 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆127Updated 2 weeks ago
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX
  ☆70Updated 2 weeks ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆132Updated 8 months ago
• stanislavfort / dissect-git-re-basin
  Replicating and dissecting the git-re-basin project in one-click-replication Colabs
  ☆36Updated 2 years ago
• helblazer811 / diffusion-visualizations
  Visualizations of the theory behind diffusion models.
  ☆75Updated 7 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"
  ☆60Updated 2 years ago
• nikhilvyas / SOAP
  ☆128Updated this week