RAFT contains fundamental widely-used algorithms and primitives for machine learning and information retrieval. The algorithms are CUDA-accelerated and form building blocks for more easily writing …

Official Implementation of "RTop-K: Ultra-Fast Row-Wise Top-K Selection for Neural Network Acceleration on GPUs"

A machine learning compiler for GPUs, CPUs, and ML accelerators

An Extensible Deep Learning Library

MLIR-based partitioning system

Backward compatible ML compute opset inspired by HLO/MHLO

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

The LLVM Project is a collection of modular and reusable compiler and toolchain technologies.

A stand-alone implementation of several NumPy dtype extensions used in machine learning.

uYou+ is a modified version of uYou (made by @MiRO92) with additional features and mainly made for non jailbroken users!

Enabling PyTorch on XLA Devices (e.g. Google TPU)

An Open Source Machine Learning Framework for Everyone

Tensors and Dynamic neural networks in Python with strong GPU acceleration

A library for training and deploying machine learning models on Amazon SageMaker

AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.

CMake driven cross-platform package manager for C/C++.

LLDB's machine interface driver

Deploy Stable Diffusion Model on Amazon SageMaker Endpont

The simplest, fastest repository for training/finetuning medium-sized GPTs.

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch

CUDA Templates for Linear Algebra Subroutines

Useful extra functionality for TensorFlow 2.x maintained by SIG-addons