aws-neuron / nki-llamaLinks
Project showing how to develop NKI kernels for Llama 3.2 1B inference
☆20Updated 6 months ago
Alternatives and similar repositories for nki-llama
Users that are interested in nki-llama are comparing it to the libraries listed below
Sorting:
- ☆58Updated last week
- ☆15Updated last week
- ☆83Updated 3 years ago
- A schedule language for large model training☆152Updated 4 months ago
- This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".☆83Updated 3 months ago
- ☆152Updated last year
- ☆63Updated last week
- Perplexity GPU Kernels☆542Updated last month
- Github mirror of trition-lang/triton repo.☆109Updated this week
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆239Updated this week
- ☆18Updated last year
- (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.☆43Updated 3 years ago
- A ChatGPT(GPT-3.5) & GPT-4 Workload Trace to Optimize LLM Serving Systems☆226Updated 5 months ago
- Compiler for Dynamic Neural Networks☆46Updated 2 years ago
- Synthesizer for optimal collective communication algorithms☆121Updated last year
- LLM serving cluster simulator☆128Updated last year
- Microsoft Collective Communication Library☆66Updated last year
- Repository for MLCommons Chakra schema and tools☆39Updated 2 years ago
- ☆167Updated last year
- An experimental CPU backend for Triton☆167Updated last month
- Bamboo is a system for running large pipeline-parallel DNNs affordably, reliably, and efficiently using spot instances.☆54Updated 3 years ago
- Supplemental materials for The ASPLOS 2025 / EuroSys 2025 Contest on Intra-Operator Parallelism for Distributed Deep Learning☆24Updated 7 months ago
- DietCode Code Release☆64Updated 3 years ago
- KernelBench: Can LLMs Write GPU Kernels? - Benchmark + Toolkit with Torch -> CUDA (+ more DSLs)☆718Updated last week
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆140Updated this week
- TACCL: Guiding Collective Algorithm Synthesis using Communication Sketches☆78Updated 2 years ago
- A resilient distributed training framework☆96Updated last year
- Cataloging released Triton kernels.☆278Updated 3 months ago
- Repository for MLCommons Chakra schema and tools☆144Updated 2 months ago
- MAGIS: Memory Optimization via Coordinated Graph Transformation and Scheduling for DNN (ASPLOS'24)☆55Updated last year