Alternatives and similar repositories for microjax

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

• valine / training-hot-swap
  Pytorch script hot swap: Change code without unloading your LLM from VRAM
  ☆125Updated 7 months ago
• ivanbelenky / RL
  R.L. methods and techniques.
  ☆199Updated last year
• DiscoGrad / DiscoGrad
  DiscoGrad - automatically differentiate across conditional branches in C++ programs
  ☆209Updated last year
• nirw4nna / dsc
  Tensor library & inference framework for machine learning
  ☆114Updated 2 months ago
• nathan-barry / tiny-diffusion
  A character-level language diffusion model trained on Tiny Shakespeare
  ☆594Updated 3 weeks ago
• koogle / mlx-playground
  time to learn mlx
  ☆40Updated 2 months ago
• mlecauchois / micrograd-cuda
  ☆249Updated last year
• slightknack / difflogic
  Train neural networks that distill into logic circuits, using JAX
  ☆63Updated 5 months ago
• utkuozbulak / pytorch-simple-diffusion
  ☆47Updated 8 months ago
• gvelesandro / constructor-theory-simulator
  ☆68Updated last month
• DebarghaG / proofofthought
  Proof of thought : LLM-based reasoning using Z3 theorem proving with multiple backend support (SMT2 and JSON DSL)
  ☆360Updated last month
• drubinstein / pokemonred_puffer
  ☆176Updated 5 months ago
• fraware / leanverifier
  Framework for specifying and proving properties—such as robustness, fairness, and interpretability—of machine learning models using Lean …
  ☆73Updated 4 months ago
• babycommando / neuralgraffiti
  Live-bending a foundation model’s output at neural network level.
  ☆271Updated 8 months ago
• akarshkumar0101 / fer
  Code for the Fractured Entangled Representation Hypothesis position paper!
  ☆214Updated last month
• trevorpogue / algebraic-nnhw
  Algebraic enhancements for GEMM & AI accelerators
  ☆282Updated 9 months ago
• dicroce / hnsw
  Heirarchical Navigable Small Worlds
  ☆101Updated 3 months ago
• google-deepmind / treescope
  An interactive HTML pretty-printer for machine learning research in IPython notebooks.
  ☆451Updated 4 months ago
• samim23 / vortexnet
  VortexNet: Neural Computing through Fluid Dynamics
  ☆46Updated 10 months ago
• joennlae / tensorli
  Absolute minimalistic implementation of a GPT-like transformer using only numpy (<650 lines).
  ☆255Updated 2 years ago
• em-llm / EM-LLM-model
  ☆245Updated 9 months ago
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆61Updated last year
• Foreseerr / TScale
  ☆199Updated 7 months ago
• westoncb / nonlinear-optics-sandbox
  ☆36Updated 3 months ago
• patrick-kidger / tinyio
  Ever used asyncio and wished you hadn't? A tiny (~300 lines) event loop for Python.
  ☆689Updated 2 months ago
• arcee-ai / pybubble
  ☆70Updated 3 weeks ago
• anishathalye / semlib
  Build data processing and data analysis pipelines that leverage the power of LLMs 🧠
  ☆242Updated last week
• deepreinforce-ai / CUDA-L2
  CUDA-L2: Surpassing cuBLAS Performance for Matrix Multiplication through Reinforcement Learning
  ☆142Updated this week
• ESHyperscale / nano-egg
  Evolution Pretraining Fully in Int Formats
  ☆111Updated last week
• statusfailed / catgrad
  a categorical deep learning compiler
  ☆205Updated 2 months ago