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,181Updated this week
• llvm / torch-mlir
  The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
  ☆1,562Updated this week
• NVIDIA / cutlass
  CUDA Templates for Linear Algebra Subroutines
  ☆7,688Updated last week
• NVIDIA / TransformerEngine
  A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper, Ada and Bla…
  ☆2,491Updated this week
• HazyResearch / ThunderKittens
  Tile primitives for speedy kernels
  ☆2,457Updated last week
• NVIDIA / FasterTransformer
  Transformer related optimization, including BERT, GPT
  ☆6,211Updated last year
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,051Updated last year
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆490Updated 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,649Updated 2 months ago
• flashinfer-ai / flashinfer
  FlashInfer: Kernel Library for LLM Serving
  ☆3,211Updated this week
• onnx / onnx-mlir
  Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure
  ☆873Updated last week
• alpa-projects / alpa
  Training and serving large-scale neural networks with auto parallelization.
  ☆3,136Updated last year
• NVIDIA / cccl
  CUDA Core Compute Libraries
  ☆1,689Updated this week
• pytorch / ao
  PyTorch native quantization and sparsity for training and inference
  ☆2,114Updated this week
• merrymercy / awesome-tensor-compilers
  A list of awesome compiler projects and papers for tensor computation and deep learning.
  ☆2,591Updated 8 months ago
• google / flax
  Flax is a neural network library for JAX that is designed for flexibility.
  ☆6,624Updated this week
• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,383Updated this week
• triton-lang / triton
  Development repository for the Triton language and compiler
  ☆15,881Updated this week
• NVIDIA / cuda-python
  CUDA Python: Performance meets Productivity
  ☆2,767Updated last week
• 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,571Updated last year
• intel / intel-extension-for-pytorch
  A Python package for extending the official PyTorch that can easily obtain performance on Intel platform
  ☆1,875Updated this week
• deepspeedai / DeepSpeed-MII
  MII makes low-latency and high-throughput inference possible, powered by DeepSpeed.
  ☆2,020Updated 2 months ago
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆703Updated this week
• flexflow / flexflow-train
  Automatically Discovering Fast Parallelization Strategies for Distributed Deep Neural Network Training
  ☆1,803Updated this week
• huggingface / safetensors
  Simple, safe way to store and distribute tensors
  ☆3,311Updated this week
• facebookresearch / fairscale
  PyTorch extensions for high performance and large scale training.
  ☆3,331Updated last month
• dmlc / dlpack
  common in-memory tensor structure
  ☆1,014Updated last week
• google / XNNPACK
  High-efficiency floating-point neural network inference operators for mobile, server, and Web
  ☆2,046Updated this week
• alibaba / BladeDISC
  BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.
  ☆873Updated 5 months ago
• NVIDIA / MatX
  An efficient C++17 GPU numerical computing library with Python-like syntax
  ☆1,329Updated last week