apple / axlearnLinks
An Extensible Deep Learning Library
☆2,261Updated 2 weeks ago
Alternatives and similar repositories for axlearn
Users that are interested in axlearn are comparing it to the libraries listed below
Sorting:
- A simple, performant and scalable Jax LLM!☆1,917Updated this week
- A PyTorch native platform for training generative AI models☆4,476Updated this week
- PyTorch native quantization and sparsity for training and inference☆2,384Updated this week
- Efficient framework-agnostic data loading☆438Updated this week
- PyTorch native post-training library☆5,517Updated this week
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆3,564Updated this week
- PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily wri…☆1,416Updated this week
- Minimalistic large language model 3D-parallelism training☆2,239Updated last month
- Run PyTorch LLMs locally on servers, desktop and mobile☆3,614Updated 3 weeks ago
- On-device AI across mobile, embedded and edge for PyTorch☆3,278Updated this week
- nanoGPT style version of Llama 3.1☆1,427Updated last year
- ☆1,004Updated 7 months ago
- Examples in the MLX framework☆7,881Updated last month
- DataComp for Language Models☆1,367Updated 3 weeks ago
- Inference Llama 2 in one file of pure 🔥☆2,118Updated last year
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,821Updated 3 months ago
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper, Ada and Bla…☆2,755Updated this week
- Tile primitives for speedy kernels☆2,767Updated 2 weeks ago
- A modern model graph visualizer and debugger☆1,318Updated this week
- Minimalistic 4D-parallelism distributed training framework for education purpose☆1,836Updated last month
- Reference implementation of the Transformer architecture optimized for Apple Neural Engine (ANE)☆2,657Updated 2 years ago
- NanoGPT (124M) in 3 minutes☆3,145Updated 2 months ago
- Reaching LLaMA2 Performance with 0.1M Dollars☆986Updated last year
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆6,113Updated last month
- Training LLMs with QLoRA + FSDP☆1,529Updated 10 months ago
- The n-gram Language Model☆1,444Updated last year
- The Autograd Engine☆637Updated last year
- [ICLR 2025] Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modeling☆912Updated 5 months ago
- ☆4,095Updated last year
- Examples using MLX Swift☆2,229Updated this week