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 9 months ago
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆362Updated this week
• AlibabaPAI / torchacc
  PyTorch distributed training acceleration framework
  ☆53Updated 3 months ago
• BBuf / tensorrt-llm-moe
  ☆33Updated 9 months ago
• bytedance / ByteMLPerf
  AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…
  ☆268Updated 3 months ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆121Updated 11 months ago
• NVIDIA / nsight-training
  Training material for Nsight developer tools
  ☆171Updated last year
• deepspeedai / DeepSpeed-Kernels
  ☆71Updated 7 months ago
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆112Updated last year
• mlcommons / logging
  MLPerf™ logging library
  ☆37Updated last month
• vllm-project / flash-attention
  Fast and memory-efficient exact attention
  ☆97Updated 2 weeks ago
• amd / ZenDNN
  ☆126Updated last week
• PaddlePaddle / CINN
  Compiler Infrastructure for Neural Networks
  ☆147Updated 2 years 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.
  ☆42Updated 8 months ago
• iree-org / iree-nvgpu
  ☆50Updated last year
• microsoft / ark
  A GPU-driven system framework for scalable AI applications
  ☆120Updated 9 months ago
• xlite-dev / HGEMM
  ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.
  ☆127Updated 6 months ago
• InternLM / turbomind
  ☆97Updated 7 months ago
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆108Updated last year
• ROCm / aotriton
  Ahead of Time (AOT) Triton Math Library
  ☆83Updated last week
• NVIDIA / nvbandwidth
  A tool for bandwidth measurements on NVIDIA GPUs.
  ☆568Updated 7 months ago
• rapidsai / ucx-py
  Python bindings for UCX
  ☆140Updated 2 months ago
• NVIDIA / online-softmax
  Benchmark code for the "Online normalizer calculation for softmax" paper
  ☆102Updated 7 years ago
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆100Updated 4 months 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 2 months ago
• pytorch / rfcs
  PyTorch RFCs (experimental)
  ☆135Updated 5 months ago
• facebookresearch / fairring
  Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …
  ☆65Updated 3 years ago
• microsoft / mscclpp
  MSCCL++: A GPU-driven communication stack for scalable AI applications
  ☆437Updated 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…
  ☆176Updated 2 weeks ago
• anyscale / llm-continuous-batching-benchmarks
  ☆122Updated last year