AMD-AGI / PrimusLinks
☆43Updated this week
Alternatives and similar repositories for Primus
Users that are interested in Primus are comparing it to the libraries listed below
Sorting:
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆79Updated 3 months ago
- Home for OctoML PyTorch Profiler☆114Updated 2 years ago
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆216Updated this week
- This repository contains the results and code for the MLPerf™ Training v0.7 benchmark.☆57Updated 2 years ago
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆393Updated 6 months ago
- ☆71Updated 8 months ago
- PyTorch RFCs (experimental)☆136Updated 6 months ago
- Dev repo for power measurement for the MLPerf™ benchmarks☆26Updated 3 months ago
- Perplexity open source garden for inference technology☆295Updated last week
- extensible collectives library in triton☆91Updated 8 months ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆162Updated 2 months ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆365Updated this week
- A tensor-aware point-to-point communication primitive for machine learning☆279Updated last month
- Pytorch Distributed native training library for LLMs/VLMs with OOTB Hugging Face support☆202Updated last week
- Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…☆63Updated 5 months ago
- Reference models for Intel(R) Gaudi(R) AI Accelerator☆169Updated 2 months ago
- torchcomms: a modern PyTorch communications API☆302Updated this week
- Write a fast kernel and run it on Discord. See how you compare against the best!☆64Updated this week
- This repository contains the results and code for the MLPerf™ Training v1.0 benchmark.☆36Updated last year
- Ship correct and fast LLM kernels to PyTorch☆126Updated last week
- Intel Gaudi's Megatron DeepSpeed Large Language Models for training☆15Updated 11 months ago
- PArametrized Recommendation and Ai Model benchmark is a repository for development of numerous uBenchmarks as well as end to end nets for…☆154Updated last week
- Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)☆456Updated last week
- oneCCL Bindings for Pytorch* (deprecated)☆103Updated last month
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆239Updated this week
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆133Updated this week
- Applied AI experiments and examples for PyTorch☆311Updated 3 months ago
- Easy and lightning fast training of 🤗 Transformers on Habana Gaudi processor (HPU)☆201Updated last week
- Open Source Continuous Inference Benchmarking - GB200 NVL72 vs MI355X vs B200 vs H200 vs MI325X & soon™ TPUv6e/v7/Trainium2/3/GB300 NVL72…☆392Updated this week
- Optimized primitives for collective multi-GPU communication☆25Updated last year