aws-neuron / nki-llamaLinks
Project showing how to develop NKI kernels for Llama 3.2 1B inference
☆19Updated 3 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:
- ☆47Updated last week
- ☆14Updated last week
- ☆60Updated this week
- A schedule language for large model training☆149Updated 2 weeks ago
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆214Updated this week
- KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA problems☆546Updated last week
- ☆115Updated 8 months ago
- Github mirror of trition-lang/triton repo.☆54Updated this week
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆209Updated 2 weeks ago
- ☆81Updated 2 years ago
- Perplexity GPU Kernels☆451Updated last month
- ☆234Updated 2 weeks ago
- DietCode Code Release☆65Updated 3 years ago
- MLIR-based partitioning system☆125Updated this week
- Collection of kernels written in Triton language☆153Updated 5 months ago
- Microsoft Collective Communication Library☆67Updated 9 months ago
- QuickReduce is a performant all-reduce library designed for AMD ROCm that supports inline compression.☆32Updated last week
- ☆37Updated last month
- LeetGPU Challenges☆61Updated this week
- ☆229Updated last year
- ☆23Updated 2 weeks ago
- A ChatGPT(GPT-3.5) & GPT-4 Workload Trace to Optimize LLM Serving Systems☆200Updated last month
- PArametrized Recommendation and Ai Model benchmark is a repository for development of numerous uBenchmarks as well as end to end nets for…☆149Updated last week
- A library to analyze PyTorch traces.☆406Updated 2 weeks ago
- An experimental CPU backend for Triton☆146Updated 3 months ago
- Cataloging released Triton kernels.☆252Updated 7 months ago
- Applied AI experiments and examples for PyTorch☆293Updated 2 weeks ago
- extensible collectives library in triton☆88Updated 5 months ago
- A resilient distributed training framework☆94Updated last year
- ☆75Updated 4 years ago