Chengji Yao

The LLVM Project is a collection of modular and reusable compiler and toolchain technologies. Note: the repository does not accept github pull requests at this moment. Please submit your patches at http://reviews.llvm.org.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

A high-throughput and memory-efficient inference and serving engine for LLMs

TPU inference for vLLM, with unified JAX and PyTorch support.

vLLM TPU Benchmark infra

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

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

Enabling PyTorch on Google TPU

Open deep learning compiler stack for cpu, gpu and specialized accelerators

ByteIR

Backward compatible ML compute opset inspired by HLO/MHLO

Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure

No description.

Open standard for machine learning interoperability

MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架

TiDB is an open source distributed HTAP database compatible with the MySQL protocol

Running BERT without Padding

An Open Source Machine Learning Framework for Everyone

Microsoft Cognitive Toolkit (CNTK), an open source deep-learning toolkit

Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more