google / tsl
☆94Updated this week
Alternatives and similar repositories for tsl:
Users that are interested in tsl are comparing it to the libraries listed below
- Starlark implementation of bazel rules for CUDA.☆99Updated this week
- MLIR-based partitioning system☆73Updated this week
- A lightweight memory allocator for hardware-accelerated machine learning☆143Updated 7 months ago
- Stores documents and resources used by the OpenXLA developer community☆117Updated 7 months ago
- ☆408Updated this week
- ☆81Updated last week
- A sandbox for quick iteration and experimentation on projects related to IREE, MLIR, and LLVM☆56Updated last month
- TPP experimentation on MLIR for linear algebra☆121Updated last week
- ☆138Updated this week
- ROCm Thrust - run Thrust dependent software on AMD GPUs☆106Updated this week
- Unified Collective Communication Library☆237Updated this week
- Pybind11 bindings for the Abseil C++ Common Libraries☆28Updated 3 weeks ago
- oneAPI Collective Communications Library (oneCCL)☆225Updated 2 weeks ago
- ☆68Updated this week
- Open source cross-platform compiler for compute-intensive loops used in AI algorithms, from Microsoft Research☆110Updated last year
- AMD’s C++ library for accelerating tensor primitives☆39Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆314Updated this week
- C++ Rules for Bazel☆203Updated this week
- Intel® Extension for MLIR. A staging ground for MLIR dialects and tools for Intel devices using the MLIR toolchain.☆129Updated last week
- ☆309Updated 3 months ago
- ROCm SPARSE marshalling library☆67Updated last week
- Tests and benchmarks for cudnn (and in the future, other nvidia libraries)☆53Updated 4 years ago
- Experiments and prototypes associated with IREE or MLIR☆50Updated 7 months ago
- ☆70Updated 2 weeks ago
- Bazel wrapper around the pybind11 repository☆109Updated 5 months ago
- ☆524Updated last week
- ☆16Updated 2 years ago
- A language and compiler for irregular tensor programs.☆139Updated 3 months ago
- MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.☆129Updated last year
- fb303 is a core set of thrift functions that provide a common mechanism for querying stats and other information from a service.☆79Updated this week