aws-neuron / nki-samplesLinks
☆47Updated this week
Alternatives and similar repositories for nki-samples
Users that are interested in nki-samples are comparing it to the libraries listed below
Sorting:
- ☆60Updated last week
- Project showing how to develop NKI kernels for Llama 3.2 1B inference☆19Updated 4 months ago
- A schedule language for large model training☆151Updated last month
- ☆39Updated 9 months ago
- extensible collectives library in triton☆88Updated 6 months ago
- ☆15Updated last week
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆213Updated last week
- ☆112Updated last year
- ☆90Updated 10 months ago
- oneCCL Bindings for Pytorch*☆102Updated last month
- MLIR-based partitioning system☆135Updated this week
- An experimental CPU backend for Triton (https//github.com/openai/triton)☆45Updated last month
- TORCH_LOGS parser for PT2☆61Updated 2 weeks ago
- ☆23Updated last month
- Home for OctoML PyTorch Profiler☆114Updated 2 years ago
- (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.☆40Updated 2 years ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆318Updated last week
- ☆240Updated last week
- ☆28Updated 8 months ago
- ☆177Updated last year
- Reference models for Intel(R) Gaudi(R) AI Accelerator☆166Updated last week
- Collection of kernels written in Triton language☆155Updated 5 months ago
- This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".☆48Updated last week
- KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA problems☆581Updated 2 weeks ago
- Distributed preprocessing and data loading for language datasets☆39Updated last year
- ☆41Updated last year
- Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…☆62Updated 3 months ago
- Applied AI experiments and examples for PyTorch☆296Updated last month
- Research and development for optimizing transformers☆130Updated 4 years ago
- Example of binding a TF32 CUTLASS GEMM kernel to PyTorch☆12Updated last year