google-ai-edge / model-explorerLinks
A modern model graph visualizer and debugger
☆1,324Updated this week
Alternatives and similar repositories for model-explorer
Users that are interested in model-explorer are comparing it to the libraries listed below
Sorting:
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,824Updated 4 months ago
- PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily wri…☆1,420Updated this week
- llama3.np is a pure NumPy implementation for Llama 3 model.☆989Updated 6 months ago
- MobileLLM Optimizing Sub-billion Parameter Language Models for On-Device Use Cases. In ICML 2024.☆1,382Updated 6 months ago
- PyTorch native quantization and sparsity for training and inference☆2,464Updated this week
- Open weights language model from Google DeepMind, based on Griffin.☆653Updated 4 months ago
- Puzzles for learning Triton☆2,074Updated 11 months ago
- Tile primitives for speedy kernels☆2,838Updated last week
- Deep learning for dummies. All the practical details and useful utilities that go into working with real models.☆820Updated 3 months ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆580Updated 2 months ago
- A complete end-to-end pipeline for LLM interpretability with sparse autoencoders (SAEs) using Llama 3.2, written in pure PyTorch and full…☆625Updated 7 months ago
- Best practices & guides on how to write distributed pytorch training code☆526Updated last week
- LLM Analytics☆692Updated last year
- On-device AI across mobile, embedded and edge for PyTorch☆3,374Updated last week
- A simple, performant and scalable Jax LLM!☆1,942Updated this week
- Official implementation of Half-Quadratic Quantization (HQQ)☆884Updated last week
- TinyChatEngine: On-Device LLM Inference Library☆906Updated last year
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆448Updated 2 months ago
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆666Updated last week
- Distributed Training Over-The-Internet☆963Updated 2 weeks ago
- [ICLR 2025] Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modeling☆917Updated 6 months ago
- The Tensor (or Array)☆451Updated last year
- ☆542Updated last year
- Speed up model training by fixing data loading.☆552Updated last week
- Felafax is building AI infra for non-NVIDIA GPUs☆568Updated 9 months ago
- Schedule-Free Optimization in PyTorch☆2,228Updated 5 months ago
- GPU programming related news and material links☆1,746Updated last month
- Official repository for our work on micro-budget training of large-scale diffusion models.☆1,521Updated 9 months ago
- UNet diffusion model in pure CUDA☆651Updated last year
- prime is a framework for efficient, globally distributed training of AI models over the internet.☆840Updated 5 months ago