AMD-AGI / PrimusLinks
☆72Updated 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 last month
- Perplexity open source garden for inference technology☆359Updated last month
- extensible collectives library in triton☆95Updated 10 months ago
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆262Updated this week
- Ship correct and fast LLM kernels to PyTorch☆140Updated 3 weeks ago
- Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)☆475Updated this week
- Write a fast kernel and run it on Discord. See how you compare against the best!☆71Updated this week
- NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process com…☆462Updated last month
- Perplexity GPU Kernels☆560Updated 3 months ago
- Applied AI experiments and examples for PyTorch☆315Updated 5 months ago
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆404Updated last month
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆168Updated this week
- ☆286Updated this week
- TPU inference for vLLM, with unified JAX and PyTorch support.☆228Updated this week
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆219Updated last week
- torchcomms: a modern PyTorch communications API☆327Updated this week
- ArcticInference: vLLM plugin for high-throughput, low-latency inference☆391Updated this week
- AI Tensor Engine for ROCm☆351Updated this week
- Evaluating Large Language Models for CUDA Code Generation ComputeEval is a framework designed to generate and evaluate CUDA code from Lar…☆96Updated last month
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆380Updated this week
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆324Updated this week
- Fast low-bit matmul kernels in Triton☆427Updated last week
- ☆71Updated 10 months ago
- TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels☆194Updated this week
- Open Source Continuous Inference Benchmarking - GB200 NVL72 vs MI355X vs B200 vs GB300 NVL72 vs H100 & soon™ TPUv6e/v7/Trainium2/3- DeepS…☆445Updated this week
- Home for OctoML PyTorch Profiler☆113Updated 2 years ago
- Cataloging released Triton kernels.☆292Updated 5 months ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆739Updated this week
- Github mirror of trition-lang/triton repo.☆128Updated this week
- Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…☆64Updated 7 months ago