apple / axlearnLinks
An Extensible Deep Learning Library
☆2,095Updated this week
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,786Updated this week
- A PyTorch native platform for training generative AI models☆3,933Updated this week
- Run PyTorch LLMs locally on servers, desktop and mobile☆3,593Updated last month
- DataComp for Language Models☆1,309Updated 3 months ago
- Examples in the MLX framework☆7,524Updated last week
- Simple, safe way to store and distribute tensors☆3,311Updated this week
- Efficient framework-agnostic data loading☆427Updated 2 weeks ago
- Reference implementation of the Transformer architecture optimized for Apple Neural Engine (ANE)☆2,633Updated 2 years ago
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆3,280Updated this week
- Accessible large language models via k-bit quantization for PyTorch.☆7,142Updated this week
- Minimalistic large language model 3D-parallelism training☆1,926Updated last week
- Thunder gives you PyTorch models superpowers for training and inference. Unlock out-of-the-box optimizations for performance, memory and …☆1,365Updated this week
- MII makes low-latency and high-throughput inference possible, powered by DeepSpeed.☆2,020Updated 2 months ago
- A modern model graph visualizer and debugger☆1,258Updated this week
- ☆864Updated last year
- Training LLMs with QLoRA + FSDP☆1,485Updated 7 months ago
- 🚀 Accelerate inference and training of 🤗 Transformers, Diffusers, TIMM and Sentence Transformers with easy to use hardware optimization…☆2,942Updated this week
- Tile primitives for speedy kernels☆2,457Updated last week
- PyTorch native post-training library☆5,273Updated this week
- Open weights language model from Google DeepMind, based on Griffin.☆641Updated 2 weeks ago
- Multi-LoRA inference server that scales to 1000s of fine-tuned LLMs☆3,024Updated last month
- PyTorch native quantization and sparsity for training and inference☆2,114Updated this week
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆5,993Updated 2 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,491Updated this week
- Tools for merging pretrained large language models.☆5,829Updated this week
- MLX: An array framework for Apple silicon☆21,057Updated this week
- The official PyTorch implementation of Google's Gemma models☆5,481Updated 3 weeks ago
- MobileLLM Optimizing Sub-billion Parameter Language Models for On-Device Use Cases. In ICML 2024.☆1,301Updated 2 months ago
- [MLSys 2024 Best Paper Award] AWQ: Activation-aware Weight Quantization for LLM Compression and Acceleration☆3,081Updated last week
- Gemma open-weight LLM library, from Google DeepMind☆3,395Updated last week