NVIDIA / atex

Related projects: ⓘ

• microsoft / ark
  A GPU-driven system framework for scalable AI applications
  ☆103Updated this week
• facebookresearch / FAMBench
  Benchmarks to capture important workloads.
  ☆28Updated 3 months ago
• microsoft / onnxruntime-tvm
  Open deep learning compiler stack for cpu, gpu and specialized accelerators
  ☆33Updated last year
• bytedance / ByteMLPerf
  AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…
  ☆188Updated 3 weeks ago
• amd / ZenDNN
  ☆80Updated 3 months ago
• microsoft / torchy
  A tracing JIT for PyTorch
  ☆18Updated 2 years ago
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆81Updated 2 months ago
• HabanaAI / SynapseAI_Core
  SynapseAI Core is a reference implementation of the SynapseAI API running on Habana Gaudi
  ☆36Updated last year
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆93Updated last week
• openxla / openxla-nvgpu
  ☆48Updated 6 months ago
• NVIDIA-Merlin / HierarchicalKV
  HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…
  ☆126Updated 3 weeks ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆92Updated last year
• NVIDIA / nvbandwidth
  A tool for bandwidth measurements on NVIDIA GPUs.
  ☆285Updated 3 months ago
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆250Updated this week
• microsoft / DeepSpeed-Kernels
  ☆50Updated 3 months ago
• ROCm / rocm_bandwidth_test
  Bandwidth test for ROCm
  ☆45Updated this week
• microsoft / dist-ir
  An IR for efficiently simulating distributed ML computation.
  ☆24Updated 8 months ago
• Harry-Chen / InfMoE
  Inference framework for MoE layers based on TensorRT with Python binding
  ☆41Updated 3 years ago
• NVIDIA / nsight-training
  Training material for Nsight developer tools
  ☆125Updated last month
• Intel-tensorflow / tensorflow
  Computation using data flow graphs for scalable machine learning
  ☆65Updated this week
• triton-inference-server / common
  Common source, scripts and utilities shared across all Triton repositories.
  ☆62Updated last week
• triton-inference-server / pytorch_backend
  The Triton backend for the PyTorch TorchScript models.
  ☆117Updated last week
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆81Updated 2 months ago
• vllm-project / flash-attention
  Fast and memory-efficient exact attention
  ☆20Updated 2 weeks ago
• intel / torch-ccl
  oneCCL Bindings for Pytorch*
  ☆83Updated last week
• ROCm / triton
  Development repository for the Triton language and compiler
  ☆86Updated this week
• triton-lang / triton-cpu
  An experimental CPU backend for Triton
  ☆36Updated last week
• NVIDIA / NVTX
  The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…
  ☆279Updated last week
• ROCm / hipBLASLt
  hipBLASLt is a library that provides general matrix-matrix operations with a flexible API and extends functionalities beyond a traditiona…
  ☆48Updated this week
• microsoft / mscclpp
  MSCCL++: A GPU-driven communication stack for scalable AI applications
  ☆233Updated this week