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
• bytedance / ByteMLPerf
  AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…
  ☆265Updated 2 months ago
• NVIDIA / nsight-training
  Training material for Nsight developer tools
  ☆170Updated last year
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆112Updated last year
• AlibabaPAI / torchacc
  PyTorch distributed training acceleration framework
  ☆53Updated 2 months ago
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆358Updated this week
• BBuf / tensorrt-llm-moe
  ☆33Updated 8 months ago
• PaddlePaddle / CINN
  Compiler Infrastructure for Neural Networks
  ☆147Updated 2 years ago
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆107Updated last year
• amd / ZenDNN
  ☆126Updated this week
• iree-org / iree-nvgpu
  ☆50Updated last year
• NVlabs / cub
  THIS REPOSITORY HAS MOVED TO github.com/nvidia/cub, WHICH IS AUTOMATICALLY MIRRORED HERE.
  ☆84Updated last year
• NVIDIA-Merlin / HierarchicalKV
  HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…
  ☆175Updated this week
• OpenPPL / CuAssembler
  An unofficial cuda assembler, for all generations of SASS, hopefully ：）
  ☆83Updated 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.
  ☆41Updated 8 months ago
• uxlfoundation / oneCCL
  oneAPI Collective Communications Library (oneCCL)
  ☆245Updated last week
• TiledTensor / TiledCUDA
  We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …
  ☆186Updated 9 months ago
• Harry-Chen / InfMoE
  Inference framework for MoE layers based on TensorRT with Python binding
  ☆41Updated 4 years ago
• rapidsai / ucx-py
  Python bindings for UCX
  ☆140Updated last month
• 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
• microsoft / ark
  A GPU-driven system framework for scalable AI applications
  ☆119Updated 8 months ago
• MLIR-China / mlir-playground
  Play with MLIR right in your browser
  ☆137Updated 2 years ago
• billmuch / matmul_perf_test
  ☆14Updated 3 years ago
• Oneflow-Inc / serving
  OneFlow Serving
  ☆20Updated 6 months ago
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆120Updated 11 months ago
• xlite-dev / HGEMM
  ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.
  ☆124Updated 5 months ago
• deepspeedai / DeepSpeed-Kernels
  ☆71Updated 7 months ago
• bdhirsh / pytorch_open_registration_example
  Example of using pytorch's open device registration API
  ☆30Updated 3 years ago
• FlagOpen / FlagCX
  ☆91Updated this week
• Tencent / WeChat-TFCC
  ☆127Updated 4 years ago