NVIDIA / cuEmbedLinks
CUDA Embedding Lookup Kernel Library
☆41Updated 3 months ago
Alternatives and similar repositories for cuEmbed
Users that are interested in cuEmbed are comparing it to the libraries listed below
Sorting:
- Ship correct and fast LLM kernels to PyTorch☆140Updated 3 weeks ago
- NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process com…☆461Updated last month
- extensible collectives library in triton☆95Updated 10 months ago
- Evaluating Large Language Models for CUDA Code Generation ComputeEval is a framework designed to generate and evaluate CUDA code from Lar…☆96Updated 3 weeks ago
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆164Updated last week
- Helpful kernel tutorials and examples for tile-based GPU programming☆630Updated this week
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆324Updated this week
- Triton-based Symmetric Memory operators and examples☆81Updated 3 weeks ago
- ☆104Updated last year
- ☆286Updated this week
- TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels☆194Updated this week
- QuickReduce is a performant all-reduce library designed for AMD ROCm that supports inline compression.☆36Updated 5 months ago
- Fast low-bit matmul kernels in Triton☆427Updated this week
- TPU inference for vLLM, with unified JAX and PyTorch support.☆228Updated this week
- Cataloging released Triton kernels.☆292Updated 4 months ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆739Updated this week
- ☆159Updated last year
- torchcomms: a modern PyTorch communications API☆327Updated this week
- Collection of kernels written in Triton language☆178Updated last week
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆106Updated 7 months ago
- This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".☆97Updated 4 months ago
- Home for OctoML PyTorch Profiler☆113Updated 2 years ago
- Low overhead tracing library and trace visualizer for pipelined CUDA kernels☆130Updated 2 months ago
- ☆259Updated last year
- An experimental CPU backend for Triton☆174Updated 2 months ago
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆255Updated last week
- Nsight Python is a Python kernel profiling interface based on NVIDIA Nsight Tools☆99Updated last week
- Autonomous GPU Kernel Generation via Deep Agents☆228Updated this week
- DeeperGEMM: crazy optimized version☆73Updated 9 months ago
- Github mirror of trition-lang/triton repo.☆129Updated last week