Supercharge huggingface transformers with model parallelism.
☆78Jul 23, 2025Updated 7 months ago
Alternatives and similar repositories for magix
Users that are interested in magix are comparing it to the libraries listed below
Sorting:
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Jun 5, 2025Updated 9 months ago
- Track and Collaborate on ML & AI Experiments.☆44Mar 10, 2025Updated 11 months ago
- Rust implementation of Surya☆65Mar 1, 2025Updated last year
- Efficient vector database for hundred millions of embeddings.☆212May 17, 2024Updated last year
- Jax like function transformation engine but micro, microjax☆34Oct 25, 2024Updated last year
- Benchmarking Mobile Device Control Agents across Diverse Configurations (ICLR 2024 workshop GenAI4DM spotlight presentation; CoLLAs 2025)☆35Jul 21, 2025Updated 7 months ago
- Code repo for "Model-Generated Pretraining Signals Improves Zero-Shot Generalization of Text-to-Text Transformers" (ACL 2023)☆22Nov 1, 2023Updated 2 years ago
- A high-throughput and memory-efficient inference and serving engine for LLMs☆267Dec 4, 2025Updated 3 months ago
- Easy to use and open-source unknown stealer☆22Jul 24, 2023Updated 2 years ago
- Utilities for efficient fine-tuning, inference and evaluation of code generation models☆21Oct 3, 2023Updated 2 years ago
- This is the official implementation for our ACL 2024 paper: "Causal Estimation of Memorisation Profiles".☆24Mar 25, 2025Updated 11 months ago
- Interactive coding assistant for data scientists and machine learning developers, empowered by large language models.☆99Oct 8, 2024Updated last year
- Repository containing the SPIN experiments on the DIBT 10k ranked prompts☆23Mar 12, 2024Updated last year
- The official GitHub page for the survey paper "A Survey on Data Augmentation in Large Model Era"☆132Jul 10, 2024Updated last year
- Train Dense Passage Retriever (DPR) with a single GPU☆136Jun 16, 2021Updated 4 years ago
- ☆11May 12, 2023Updated 2 years ago
- A scalable benchmark for state representation learning in visual reinforcement learning.☆16Jun 23, 2025Updated 8 months ago
- Python interface to cd-hit☆10Feb 26, 2019Updated 7 years ago
- Label shift estimation for transfer difficulty with Familiarity.☆10Feb 4, 2025Updated last year
- Baguetter is a flexible, efficient, and hackable search engine library implemented in Python. It's designed for quickly benchmarking, imp…☆206Aug 31, 2024Updated last year
- An open source replication of the stawberry method that leverages Monte Carlo Search with PPO and or DPO☆29Feb 23, 2026Updated last week
- MIO: A Foundation Model on Multimodal Tokens☆34Dec 13, 2024Updated last year
- Tree-based indexes for neural-search☆31Mar 4, 2024Updated 2 years ago
- code for training & evaluating Contextual Document Embedding models☆201May 14, 2025Updated 9 months ago
- ☆10Dec 12, 2023Updated 2 years ago
- This is an example of creating an AI agent with flowchart☆12Jul 22, 2024Updated last year
- Repository for the paper: "TiC-LM: A Web-Scale Benchmark for Time-Continual LLM Pretraining" ACL Oral 2025☆21Feb 27, 2026Updated last week
- Authors' implementation of the paper "Equivariant Networks for Pixelized Spheres" published at ICML 2021.☆12Feb 23, 2022Updated 4 years ago
- A pipeline parallel training script for LLMs.☆167Apr 30, 2025Updated 10 months ago
- An efficent implementation of the method proposed in "The Era of 1-bit LLMs"☆155Oct 15, 2024Updated last year
- ☆19Apr 28, 2021Updated 4 years ago
- Course materials for 11-767☆13Nov 10, 2022Updated 3 years ago
- ☆12Oct 23, 2022Updated 3 years ago
- Evaluate state-of-the-art sparse embedding models on the LIMIT dataset (`limit-small` and `limit`) from google's paper `On the Theoretica…☆15Sep 4, 2025Updated 6 months ago
- ☆13Jul 14, 2024Updated last year
- seqax = sequence modeling + JAX☆186Jul 23, 2025Updated 7 months ago
- Cookbook for Crafting Good Code☆57Mar 19, 2024Updated last year
- Grep for YAML and JSON files☆109Feb 3, 2026Updated last month
- ☆19Sep 16, 2025Updated 5 months ago