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:
- NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process com…☆461Updated last month
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆164Updated this week
- Fast low-bit matmul kernels in Triton☆427Updated this week
- oneCCL Bindings for Pytorch* (deprecated)☆104Updated last month
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆380Updated this week
- ☆286Updated this week
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆324Updated this week
- Cataloging released Triton kernels.☆291Updated 4 months ago
- TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels☆194Updated this week
- ☆102Updated last year
- Official Problem Sets / Reference Kernels for the GPU MODE Leaderboard!☆200Updated last week
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆739Updated this week
- Helpful kernel tutorials and examples for tile-based GPU programming☆630Updated this week
- extensible collectives library in triton☆95Updated 10 months ago
- A Python library transfers PyTorch tensors between CPU and NVMe☆125Updated last year
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆127Updated last year
- Benchmark code for the "Online normalizer calculation for softmax" paper☆105Updated 7 years ago
- ☆159Updated last year
- ☆259Updated last year
- A lightweight design for computation-communication overlap.☆219Updated 2 weeks ago
- TPU inference for vLLM, with unified JAX and PyTorch support.☆228Updated this week
- Triton-based Symmetric Memory operators and examples☆81Updated 3 weeks ago
- Home for OctoML PyTorch Profiler☆113Updated 2 years ago
- kernels, of the mega variety☆665Updated last week
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆82Updated last year
- A Quirky Assortment of CuTe Kernels☆781Updated this week
- Github mirror of trition-lang/triton repo.☆129Updated last week
- Applied AI experiments and examples for PyTorch☆315Updated 5 months ago
- QuickReduce is a performant all-reduce library designed for AMD ROCm that supports inline compression.☆36Updated 5 months ago
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆63Updated 2 weeks ago