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 7 months ago
• bytedance / ByteMLPerf
  AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…
  ☆263Updated last month
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆354Updated last week
• AlibabaPAI / torchacc
  PyTorch distributed training acceleration framework
  ☆52Updated last month
• vllm-project / flash-attention
  Fast and memory-efficient exact attention
  ☆93Updated last week
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆110Updated last year
• Ascend / torchair
  ☆18Updated this week
• microsoft / ark
  A GPU-driven system framework for scalable AI applications
  ☆118Updated 7 months ago
• Harry-Chen / InfMoE
  Inference framework for MoE layers based on TensorRT with Python binding
  ☆41Updated 4 years ago
• BBuf / tensorrt-llm-moe
  ☆32Updated 7 months ago
• PaddlePaddle / CINN
  Compiler Infrastructure for Neural Networks
  ☆147Updated 2 years ago
• deepspeedai / DeepSpeed-Kernels
  ☆74Updated 5 months ago
• NVIDIA / nsight-training
  Training material for Nsight developer tools
  ☆164Updated last year
• rapidsai / ucx-py
  Python bindings for UCX
  ☆140Updated this week
• OpenPPL / CuAssembler
  An unofficial cuda assembler, for all generations of SASS, hopefully ：）
  ☆84Updated 2 years ago
• xlite-dev / HGEMM
  ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.
  ☆112Updated 4 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…
  ☆170Updated 2 weeks ago
• Intel-tensorflow / tensorflow
  Computation using data flow graphs for scalable machine learning
  ☆68Updated this week
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆105Updated last year
• 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 3 weeks ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆121Updated 9 months ago
• DeepLink-org / ditorch
  ☆23Updated 8 months ago
• NVlabs / cub
  THIS REPOSITORY HAS MOVED TO github.com/nvidia/cub, WHICH IS AUTOMATICALLY MIRRORED HERE.
  ☆84Updated last year
• DeepRec-AI / HybridBackend
  A high-performance framework for training wide-and-deep recommender systems on heterogeneous cluster
  ☆158Updated last year
• microsoft / mscclpp
  MSCCL++: A GPU-driven communication stack for scalable AI applications
  ☆417Updated this week
• 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 6 months ago
• NVIDIA / nvbandwidth
  A tool for bandwidth measurements on NVIDIA GPUs.
  ☆532Updated 5 months ago
• MLIR-China / mlir-playground
  Play with MLIR right in your browser
  ☆136Updated 2 years ago
• bdhirsh / pytorch_open_registration_example
  Example of using pytorch's open device registration API
  ☆30Updated 2 years ago
• FlagOpen / FlagCX
  ☆88Updated this week