Alternatives and similar repositories for xla

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

• iree-org / iree
  A retargetable MLIR-based machine learning compiler and runtime toolkit.
  ☆3,475Updated this week
• llvm / torch-mlir
  The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
  ☆1,684Updated this week
• facebookincubator / AITemplate
  AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…
  ☆4,695Updated last month
• HazyResearch / ThunderKittens
  Tile primitives for speedy kernels
  ☆2,955Updated this week
• NVIDIA / TransformerEngine
  A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on H…
  ☆2,954Updated last week
• triton-lang / triton
  Development repository for the Triton language and compiler
  ☆17,668Updated last week
• NVIDIA / cutlass
  CUDA Templates and Python DSLs for High-Performance Linear Algebra
  ☆8,865Updated this week
• NVIDIA / FasterTransformer
  Transformer related optimization, including BERT, GPT
  ☆6,355Updated last year
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆573Updated 2 weeks ago
• google / flax
  Flax is a neural network library for JAX that is designed for flexibility.
  ☆6,934Updated last week
• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,487Updated this week
• NVIDIA / cccl
  CUDA Core Compute Libraries
  ☆2,047Updated this week
• google / XNNPACK
  High-efficiency floating-point neural network inference operators for mobile, server, and Web
  ☆2,184Updated this week
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,067Updated last year
• huggingface / safetensors
  Simple, safe way to store and distribute tensors
  ☆3,528Updated 2 weeks ago
• onnx / onnx-mlir
  Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure
  ☆937Updated last week
• mirage-project / mirage
  Mirage Persistent Kernel: Compiling LLMs into a MegaKernel
  ☆1,973Updated last week
• pytorch / ao
  PyTorch native quantization and sparsity for training and inference
  ☆2,531Updated last week
• apache / tvm
  Open Machine Learning Compiler Framework
  ☆12,863Updated this week
• flashinfer-ai / flashinfer
  FlashInfer: Kernel Library for LLM Serving
  ☆4,136Updated this week
• NVIDIA / nccl
  Optimized primitives for collective multi-GPU communication
  ☆4,274Updated 3 weeks ago
• NVIDIA / cuda-python
  CUDA Python: Performance meets Productivity
  ☆3,053Updated last week
• dmlc / dlpack
  common in-memory tensor structure
  ☆1,106Updated last month
• ELS-RD / kernl
  Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackab…
  ☆1,585Updated last year
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆736Updated 2 months ago
• intel / neural-compressor
  SOTA low-bit LLM quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4) & sparsity; leading model compression techniques on PyTorch, TensorFlow, …
  ☆2,537Updated this week
• facebookresearch / fairscale
  PyTorch extensions for high performance and large scale training.
  ☆3,386Updated 7 months ago
• google-deepmind / alphatensor
  ☆2,795Updated last year
• tenstorrent / tt-metal
  TT-NN operator library, and TT-Metalium low level kernel programming model.
  ☆1,265Updated last week
• pytorch / xla
  Enabling PyTorch on XLA Devices (e.g. Google TPU)
  ☆2,709Updated last week