apple / axlearn
An Extensible Deep Learning Library
☆1,874Updated this week
Related projects ⓘ
Alternatives and complementary repositories for axlearn
- A native PyTorch Library for large model training☆2,623Updated this week
- PyTorch native quantization and sparsity for training and inference☆1,585Updated this week
- Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enables using different hardware executors a…☆1,199Updated this week
- A simple, performant and scalable Jax LLM!☆1,532Updated this week
- PyTorch native finetuning library☆4,336Updated this week
- CoreNet: A library for training deep neural networks☆6,983Updated last month
- A modern model graph visualizer and debugger☆1,058Updated this week
- ☆892Updated last month
- Tile primitives for speedy kernels☆1,658Updated this week
- Efficient framework-agnostic data loading☆380Updated 2 months ago
- Examples using MLX Swift☆1,023Updated this week
- Examples in the MLX framework☆6,235Updated last week
- ☆1,022Updated 10 months ago
- 4M: Massively Multimodal Masked Modeling☆1,607Updated last month
- On-device AI across mobile, embedded and edge for PyTorch☆2,191Updated this week
- Multi-LoRA inference server that scales to 1000s of fine-tuned LLMs☆2,205Updated this week
- Tools for merging pretrained large language models.☆4,816Updated 2 weeks ago
- ☆448Updated 7 months ago
- SGLang is a fast serving framework for large language models and vision language models.☆6,127Updated this week
- ☆2,746Updated 2 months ago
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs…☆1,979Updated this week
- MobileLLM Optimizing Sub-billion Parameter Language Models for On-Device Use Cases. In ICML 2024.☆1,155Updated 2 weeks ago
- This repository provides the code and model checkpoints of the research paper: Scalable Pre-training of Large Autoregressive Image Model…☆697Updated 6 months ago
- ☆4,035Updated 5 months ago
- 🚀 Accelerate training and inference of 🤗 Transformers and 🤗 Diffusers with easy to use hardware optimization tools☆2,576Updated this week
- GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projection☆1,435Updated 3 weeks ago
- Minimalistic large language model 3D-parallelism training☆1,260Updated this week
- Run PyTorch LLMs locally on servers, desktop and mobile☆3,383Updated this week
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆2,710Updated this week