aws-neuron / nki-llamaLinks
Project showing how to develop NKI kernels for Llama 3.2 1B inference
☆19Updated 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:
- ☆56Updated 3 weeks ago
- ☆63Updated 2 weeks ago
- ☆15Updated last week
- A schedule language for large model training☆151Updated 3 months ago
- QuickReduce is a performant all-reduce library designed for AMD ROCm that supports inline compression.☆35Updated 3 months ago
- ☆148Updated 11 months ago
- Building the Virtuous Cycle for AI-driven LLM Systems☆93Updated this week
- LLM serving cluster simulator☆122Updated last year
- Perplexity GPU Kernels☆534Updated 3 weeks ago
- KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA (+ more DSLs)☆683Updated this week
- ☆83Updated 3 years ago
- DietCode Code Release☆64Updated 3 years ago
- A ChatGPT(GPT-3.5) & GPT-4 Workload Trace to Optimize LLM Serving Systems☆220Updated 4 months ago
- PArametrized Recommendation and Ai Model benchmark is a repository for development of numerous uBenchmarks as well as end to end nets for…☆153Updated last week
- This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".☆81Updated 2 months ago
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆239Updated this week
- ☆92Updated 3 years ago
- Github mirror of trition-lang/triton repo.☆100Updated this week
- A baseline repository of Auto-Parallelism in Training Neural Networks☆147Updated 3 years ago
- A resilient distributed training framework☆96Updated last year
- Compiler for Dynamic Neural Networks☆46Updated 2 years ago
- Supplemental materials for The ASPLOS 2025 / EuroSys 2025 Contest on Intra-Operator Parallelism for Distributed Deep Learning☆24Updated 6 months ago
- nnScaler: Compiling DNN models for Parallel Training☆120Updated 2 months ago
- Autonomous GPU Kernel Generation via Deep Agents☆172Updated this week
- (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.☆43Updated 3 years ago
- Boost hardware utilization for ML training workloads via Inter-model Horizontal Fusion☆32Updated last year
- An experimental CPU backend for Triton☆164Updated 3 weeks ago
- Latency and Memory Analysis of Transformer Models for Training and Inference☆466Updated 7 months ago
- ☆158Updated 6 months ago
- ☆45Updated last year