A high-throughput and memory-efficient inference and serving engine for LLMs
Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.
The AI Compute Platform for frontier teams. SkyPilot turns fragmented AI compute into one AI supercomputer, so frontier AI teams build custom intelligence faster.
Fast and flexible AutoML with learning guarantees.
Everything we actually know about the Apple Neural Engine (ANE)
Large-scale LLM inference engine
GPT2 for Multiple Languages, including pretrained models. GPT2 多语言支持, 15亿参数中文预训练模型
Elegant and Performant Deep Learning
Differentiable Fluid Dynamics Package
Everything you want to know about Google Cloud TPU
Implementation of a Tensor Processing Unit for embedded systems and the IoT.
Neural network-based chess engine capable of natural language commentary
OpenVision (ICCV 2025), OpenVision 2 (CVPR 2026), and OpenVision 3
JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).
DECIMER Image Transformer is a deep-learning-based tool designed for automated recognition of chemical structure images. Leveraging transformer architectures, the model converts chemical images into SMILES strings, enabling the digitization of chemical data from scanned documents, literature, and patents.
Dual Edge TPU Adapter to use it on a system with single PCIe port on m.2 A/B/E/M slot
Free TPU for FPGA with compiler supporting Pytorch/Caffe/Darknet/NCNN. An AI processor for using Xilinx FPGA to solve image classification, detection, and segmentation problem.
torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JAX-Pytorch interoperability, meaning, one can mix JAX & Pytorch syntax together when authoring ML programs, and run it in every hardware JAX can run.
Small-scale Tensor Processing Unit built on an FPGA
Add a description, image, and links to the tpu topic page so that developers can more easily learn about it.
To associate your repository with the tpu topic, visit your repo's landing page and select "manage topics."