Alternatives and similar repositories for nano-egg

Users that are interested in nano-egg are comparing it to the libraries listed below

• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated last week
• glassroom / heinsen_sequence
  Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)
  ☆98Updated last year
• ESHyperscale / HyperscaleES
  Jax Codebase for Evolutionary Strategies at the Hyperscale
  ☆188Updated last month
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆34Updated last year
• KhoomeiK / complexity-scaling
  gzip Predicts Data-dependent Scaling Laws
  ☆34Updated last year
• jfpuget / ARC-AGI-Challenge-2024
  ☆56Updated last year
• google-research / optformer
  ☆232Updated 2 weeks ago
• kmkolasinski / nano-umap
  Simplified implementation of UMAP like dimensionality reduction algorithm
  ☆53Updated last year
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated last year
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆61Updated last year
• yobibyte / report
  Because we don't want a jupyter notebook mess...
  ☆61Updated 6 months ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆135Updated last year
• LucasPrietoAl / grokking-at-the-edge-of-numerical-stability
  ☆105Updated 4 months ago
• etimush / ARC_NCA
  Repo for solving arc problems with an Neural Cellular Automata
  ☆22Updated 7 months ago
• NVlabs / Forecasting-Model-Search
  A system for automating selection and optimization of pre-trained models from the TAO Model Zoo
  ☆30Updated last year
• MilesCranmer / pysr_scaling_laws
  You should use PySR to find scaling laws. Here's an example.
  ☆33Updated 2 years ago
• irhum / hyena
  JAX/Flax implementation of the Hyena Hierarchy
  ☆34Updated 2 years ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆68Updated last year
• bhoov / hamux
  Hierarchical Associative Memory User Experience
  ☆105Updated last month
• daios-ai / mindscript
  A programming language for formal/informal computation.
  ☆42Updated 4 months ago
• okarthikb / state-space-models
  ☆28Updated last year
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆114Updated last year
• keyonvafa / world-model-evaluation
  ☆72Updated last year
• google-deepmind / neural_networks_solomonoff_induction
  Learning Universal Predictors
  ☆81Updated last year
• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆106Updated 3 weeks ago
• iliao2345 / CompressARC
  ☆208Updated 4 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆316Updated 4 months ago
• shikaiqiu / compute-better-spent
  ☆62Updated last year
• saurabhaloneai / Llama-3-From-Scratch-In-Pure-Jax
  This repository contain the simple llama3 implementation in pure jax.
  ☆70Updated 10 months ago