NVIDIA / atex

Related projects ⓘ

Alternatives and complementary repositories for atex

• triton-inference-server / common
  Common source, scripts and utilities shared across all Triton repositories.
  ☆62Updated this week
• microsoft / ark
  A GPU-driven system framework for scalable AI applications
  ☆109Updated last month
• NVIDIA / nvbandwidth
  A tool for bandwidth measurements on NVIDIA GPUs.
  ☆321Updated last month
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆84Updated 4 months ago
• bytedance / ByteMLPerf
  AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…
  ☆206Updated this week
• facebookresearch / FAMBench
  Benchmarks to capture important workloads.
  ☆28Updated 5 months ago
• vllm-project / flash-attention
  Fast and memory-efficient exact attention
  ☆30Updated 3 weeks ago
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆271Updated this week
• AlibabaPAI / torchacc
  PyTorch distributed training acceleration framework
  ☆34Updated this week
• NVIDIA-Merlin / HierarchicalKV
  HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…
  ☆133Updated 2 months ago
• NVIDIA / TensorRT-Incubator
  Experimental projects related to TensorRT
  ☆81Updated this week
• amd / ZenDNN
  ☆88Updated last week
• AlibabaPAI / FLASHNN
  ☆79Updated 2 months ago
• kaixih / tf_op_graph
  A visualization tool to display TF-Grappler optimized op graph
  ☆12Updated 2 years ago
• Intel-tensorflow / tensorflow
  Computation using data flow graphs for scalable machine learning
  ☆67Updated this week
• ROCm / rocm_bandwidth_test
  Bandwidth test for ROCm
  ☆49Updated this week
• oneapi-src / oneCCL
  oneAPI Collective Communications Library (oneCCL)
  ☆206Updated this week
• iree-org / iree-nvgpu
  ☆48Updated 8 months ago
• microsoft / triton-shared
  Shared Middle-Layer for Triton Compilation
  ☆191Updated this week
• TiledTensor / TiledCUDA
  TiledCUDA is a highly efficient kernel template library designed to elevate CUDA C’s level of abstraction for processing tiles.
  ☆156Updated this week
• mlcommons / logging
  MLPerf™ logging library
  ☆30Updated this week
• nod-ai / SHARK-ModelDev
  Unified compiler/runtime for interfacing with PyTorch Dynamo.
  ☆95Updated this week
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆98Updated 2 months ago
• HabanaAI / SynapseAI_Core
  SynapseAI Core is a reference implementation of the SynapseAI API running on Habana Gaudi
  ☆37Updated last year
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆98Updated last week
• microsoft / mscclpp
  MSCCL++: A GPU-driven communication stack for scalable AI applications
  ☆250Updated this week
• triton-inference-server / pytorch_backend
  The Triton backend for the PyTorch TorchScript models.
  ☆127Updated this week
• ROCm / roctracer
  ROCm Tracer Callback/Activity Library for Performance tracing AMD GPUs
  ☆75Updated 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…
  ☆307Updated this week
• microsoft / DeepSpeed-Kernels
  ☆55Updated 5 months ago