zhisbug / ray-scalable-ml-designLinks
Some microbenchmarks and design docs before commencement
☆12Updated 4 years ago
Alternatives and similar repositories for ray-scalable-ml-design
Users that are interested in ray-scalable-ml-design are comparing it to the libraries listed below
Sorting:
- WIP. Veloce is a low-code Ray-based parallelization library that makes machine learning computation novel, efficient, and heterogeneous.☆18Updated 3 years ago
- A collection of reproducible inference engine benchmarks☆33Updated 5 months ago
- Benchmark for machine learning model online serving (LLM, embedding, Stable-Diffusion, Whisper)☆28Updated 2 years ago
- A Ray-based data loader with per-epoch shuffling and configurable pipelining, for shuffling and loading training data for distributed tra…☆18Updated 2 years ago
- Make triton easier☆47Updated last year
- Distributed ML Optimizer☆33Updated 4 years ago
- Deadline-based hyperparameter tuning on RayTune.☆31Updated 5 years ago
- A lightweight, user-friendly data-plane for LLM training.☆28Updated 3 weeks ago
- NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference☆68Updated 9 months ago
- ☆31Updated 5 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Updated 4 months ago
- ☆10Updated last year
- DPO, but faster 🚀☆44Updated 10 months ago
- Code repository for the paper - "AdANNS: A Framework for Adaptive Semantic Search"☆65Updated last year
- FlexAttention w/ FlashAttention3 Support☆27Updated last year
- CUDA and Triton implementations of Flash Attention with SoftmaxN.☆73Updated last year
- code for paper "Accessing higher dimensions for unsupervised word translation"☆21Updated 2 years ago
- Linear Attention Sequence Parallelism (LASP)☆87Updated last year
- A memory efficient DLRM training solution using ColossalAI☆106Updated 2 years ago
- Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …☆65Updated 3 years ago
- PyTorch centric eager mode debugger☆48Updated 9 months ago
- ☆14Updated 2 years ago
- [NeurIPS 2025] Scaling Speculative Decoding with Lookahead Reasoning☆42Updated 2 weeks ago
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆64Updated 5 months ago
- Compression for Foundation Models☆35Updated 2 months ago
- A place to store reusable transformer components of my own creation or found on the interwebs☆60Updated this week
- Using FlexAttention to compute attention with different masking patterns☆45Updated last year
- Lightning In-Memory Object Store☆47Updated 3 years ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆46Updated last year
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆161Updated 2 weeks ago