awslabs / nki-autotuneLinks
☆15Updated this week
Alternatives and similar repositories for nki-autotune
Users that are interested in nki-autotune are comparing it to the libraries listed below
Sorting:
- ☆63Updated last month
- ☆53Updated this week
- Project showing how to develop NKI kernels for Llama 3.2 1B inference☆19Updated 5 months ago
- Powering AWS purpose-built machine learning chips. Blazing fast and cost effective, natively integrated into PyTorch and TensorFlow and i…☆553Updated this week
- ☆39Updated 10 months ago
- ☆22Updated last month
- ☆14Updated last year
- ☆12Updated 5 months ago
- Example code for AWS Neuron SDK developers building inference and training applications☆150Updated last week
- Collection of best practices, reference architectures, model training examples and utilities to train large models on AWS.☆366Updated this week
- A CLI tool that helps manage training jobs on the SageMaker HyperPod clusters orchestrated by Amazon EKS☆32Updated this week
- ☆13Updated last month
- ☆111Updated 10 months ago
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆232Updated this week
- ☆23Updated last year
- EFA/NCCL base AMI build Packer and CodeBuild/Pipeline files. Also base Docker build files to enable EFA/NCCL in containers☆43Updated 2 years ago
- ☆56Updated 2 weeks ago
- ☆181Updated last year
- ☆14Updated last year
- KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA (+ more DSLs)☆655Updated last week
- Scripts to customize AWS ParallelCluster☆26Updated 2 months ago
- Foundation model benchmarking tool. Run any model on any AWS platform and benchmark for performance across instance type and serving stac…☆254Updated 7 months ago
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆216Updated last week
- ☆33Updated 2 months ago
- A schedule language for large model training☆151Updated 2 months ago
- ☆545Updated last year
- ☆147Updated 10 months ago
- ☆247Updated this week
- Cataloging released Triton kernels.☆265Updated 2 months ago
- Some crazy experiments☆35Updated 2 months ago