Alternatives and similar repositories for atex

Users that are interested in atex are comparing it to the libraries listed below

• facebookresearch / FAMBench
  Benchmarks to capture important workloads.
  ☆31Updated 8 months ago
• bytedance / ByteMLPerf
  AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…
  ☆266Updated last month
• NVIDIA / nsight-training
  Training material for Nsight developer tools
  ☆166Updated last year
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆356Updated this week
• AlibabaPAI / torchacc
  PyTorch distributed training acceleration framework
  ☆52Updated last month
• rapidsai / ucx-py
  Python bindings for UCX
  ☆140Updated 3 weeks ago
• iree-org / iree-nvgpu
  ☆50Updated last year
• uxlfoundation / oneCCL
  oneAPI Collective Communications Library (oneCCL)
  ☆245Updated 2 weeks ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆121Updated 10 months ago
• NVIDIA / nvbandwidth
  A tool for bandwidth measurements on NVIDIA GPUs.
  ☆541Updated 5 months ago
• BBuf / tensorrt-llm-moe
  ☆33Updated 8 months ago
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆106Updated last year
• microsoft / dist-ir
  An IR for efficiently simulating distributed ML computation.
  ☆29Updated last year
• deepspeedai / DeepSpeed-Kernels
  ☆72Updated 6 months ago
• microsoft / ark
  A GPU-driven system framework for scalable AI applications
  ☆118Updated 8 months ago
• triton-inference-server / onnxruntime_backend
  The Triton backend for the ONNX Runtime.
  ☆162Updated 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…
  ☆172Updated last week
• NVlabs / cub
  THIS REPOSITORY HAS MOVED TO github.com/nvidia/cub, WHICH IS AUTOMATICALLY MIRRORED HERE.
  ☆84Updated last year
• PaddlePaddle / CINN
  Compiler Infrastructure for Neural Networks
  ☆147Updated 2 years ago
• microsoft / Accera
  Open source cross-platform compiler for compute-intensive loops used in AI algorithms, from Microsoft Research
  ☆109Updated 2 years ago
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆111Updated last year
• vllm-project / flash-attention
  Fast and memory-efficient exact attention
  ☆94Updated this week
• intel / intel-extension-for-deepspeed
  Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…
  ☆63Updated 3 months ago
• OpenPPL / CuAssembler
  An unofficial cuda assembler, for all generations of SASS, hopefully ：）
  ☆83Updated 2 years ago
• meta-pytorch / multipy
  torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…
  ☆181Updated last month
• microsoft / mscclpp
  MSCCL++: A GPU-driven communication stack for scalable AI applications
  ☆417Updated this week
• OpenPPL / ppl.llm.kernel.cuda
  ☆150Updated 9 months ago
• Bruce-Lee-LY / flash_attention_inference
  Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.
  ☆40Updated 7 months ago
• xlite-dev / HGEMM
  ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.
  ☆119Updated 5 months ago
• NVIDIA / NVTX
  The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…
  ☆454Updated 2 weeks ago