Alternatives and similar repositories for microml

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

• jax-ml / jax-llm-examples
  Minimal yet performant LLM examples in pure JAX
  ☆240Updated 3 weeks ago
• jax-ml / scaling-book
  Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
  ☆843Updated last week
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆306Updated last year
• google-deepmind / nanodo
  ☆289Updated last year
• rwitten / HighPerfLLMs2024
  ☆562Updated last year
• Quentin-Anthony / nanoMPI
  Simple MPI implementation for prototyping or learning
  ☆300Updated 6 months ago
• divyamakkar0 / JAXformer
  A zero-to-one guide on scaling modern transformers with n-dimensional parallelism.
  ☆115Updated last month
• erfanzar / EasyDeL
  Accelerate, Optimize performance with streamlined training and serving options with JAX.
  ☆336Updated this week
• microsoft / dion
  Dion optimizer algorithm
  ☆424Updated 3 weeks ago
• MekkCyber / TritonAcademy
  A repository to unravel the language of GPUs, making their kernel conversations easy to understand
  ☆198Updated 8 months ago
• jax-ml / bonsai
  Minimal, lightweight JAX implementations of popular models.
  ☆187Updated this week
• Quentin-Anthony / torch-profiling-tutorial
  ☆541Updated 6 months ago
• clu0 / unet.cu
  UNet diffusion model in pure CUDA
  ☆661Updated last year
• rkinas / triton-resources
  A curated list of resources for learning and exploring Triton, OpenAI's programming language for writing efficient GPU code.
  ☆457Updated 10 months ago
• MarioSieg / magnetron
  (WIP) A small but powerful, homemade PyTorch from scratch.
  ☆672Updated last week
• marin-community / levanter
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆693Updated last week
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆170Updated 6 months ago
• google / grain
  Library for reading and processing ML training data.
  ☆680Updated this week
• marin-community / marin
  Open-source framework for the research and development of foundation models.
  ☆752Updated this week
• LambdaLabsML / distributed-training-guide
  Best practices & guides on how to write distributed pytorch training code
  ☆575Updated 3 months ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆558Updated 3 weeks ago
• Keen-Technologies / physical_atari
  Platform for evaluating reinforcement learning (RL) algorithms on a physical Atari system.
  ☆145Updated 5 months ago
• huggingface / picotron_tutorial
  ☆232Updated 2 months ago
• changjonathanc / flex-nano-vllm
  FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.
  ☆334Updated 3 months ago
• kvfrans / lmpo
  ☆136Updated last month
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆63Updated 2 years ago
• srush / Autodiff-Puzzles
  ☆492Updated last year
• google / orbax
  Orbax provides common checkpointing and persistence utilities for JAX users
  ☆479Updated this week
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆382Updated this week
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆437Updated last month