Alternatives and similar repositories for Learning-JAX

Users that are interested in Learning-JAX are comparing it to the libraries listed below

• BlackHC / neural_net_checklist
  ☆215Updated last year
• jax-ml / bonsai
  Minimal, lightweight JAX implementations of popular models.
  ☆187Updated this week
• ludocomito / learning-diffusion
  A practical guide to diffusion models, implemented from scratch.
  ☆244Updated last month
• divyamakkar0 / JAXformer
  A zero-to-one guide on scaling modern transformers with n-dimensional parallelism.
  ☆115Updated last month
• che-shr-cat / Deep_Learning_with_JAX
  Notebooks for the "Deep Learning with JAX" book
  ☆168Updated 7 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆306Updated last year
• ml-gde / jaxgarden
  A collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax
  ☆22Updated 7 months ago
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆298Updated last year
• jax-ml / jax-llm-examples
  Minimal yet performant LLM examples in pure JAX
  ☆236Updated 3 weeks ago
• google / torchax
  torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JA…
  ☆175Updated this week
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆282Updated 7 months ago
• MekkCyber / TritonAcademy
  A repository to unravel the language of GPUs, making their kernel conversations easy to understand
  ☆197Updated 8 months ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated last year
• google-research / kauldron
  Modular, scalable library to train ML models
  ☆204Updated this week
• huggingface / trl-tuto
  ☆52Updated 8 months ago
• jax-ml / jax-ai-stack
  ☆265Updated this week
• Atcold / Energy-Book
  Deep Learning, an Energy Approach
  ☆241Updated 7 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆68Updated last year
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆312Updated 3 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆322Updated 5 months ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆33Updated last year
• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆98Updated 6 months ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆558Updated 3 weeks ago
• francoisfleuret / dlc
  ☆42Updated last year
• saurabhaloneai / Llama-3-From-Scratch-In-Pure-Jax
  This repository contain the simple llama3 implementation in pure jax.
  ☆71Updated 11 months ago
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆141Updated 2 years ago
• cgarciae / simple_flow_matching
  ☆24Updated last year
• AI-Hypercomputer / RecML
  ☆214Updated last week
• apple / ml-pararnn
  ☆166Updated 3 months ago
• google / metrax
  A JAX-native High Performance Eval Metrics Library
  ☆58Updated last week