Platform-Optimized Wheels ML Frameworks

Pre-compiled ML framework binaries and wheels optimized for specific hardware architectures and platforms (ARM, x86, GPU accelerators, Windows, macOS). Does NOT include general ML frameworks, source code repositories, or framework documentation.

There are 56 platform-optimized wheels frameworks tracked. 2 score above 70 (verified tier). The highest-rated is tensorflow/tensorflow at 100/100 with 194,185 stars and 21,821,835 monthly downloads. 3 of the top 10 are actively maintained.

Get all 56 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=platform-optimized-wheels&limit=20"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

#	Framework	Score	Tier	Stars	Language
1	tensorflow/tensorflow An Open Source Machine Learning Framework for Everyone	100	Verified	194,185	C++
2	microsoft/tensorwatch Debugging, monitoring and visualization for Python Machine Learning and Data Science	82	Verified	3,466	Jupyter Notebook
3	hidet-org/hidet An open-source efficient deep learning framework/compiler, written in python.	54	Established	738	Python
4	KomputeProject/kompute General purpose GPU compute framework built on Vulkan to support 1000s of...	54	Established	2,470	C++
5	NVIDIA/nvshmem NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on...	52	Established	477	C++
6	NVIDIA/DIGITS Deep Learning GPU Training System	51	Established	4,181	HTML
7	hailo-ai/hailort-drivers The Hailo PCIe driver is required for interacting with a Hailo device over...	49	Emerging	79	C
8	nod-ai/AMD-SHARK-Studio AMD-SHARK Studio -- Web UI for SHARK+IREE High Performance Machine Learning...	46	Emerging	1,454	Python
9	lhelontra/tensorflow-on-arm TensorFlow for Arm	44	Emerging	1,053	Shell
10	samjabrahams/tensorflow-on-raspberry-pi TensorFlow for Raspberry Pi	44	Emerging	2,257	Python
11	NVIDIA/DL4AGX Deep Learning tools and applications for NVIDIA AGX platforms.	43	Emerging	271	Shell
12	intel/intel-extension-for-tensorflow Intel® Extension for TensorFlow*	42	Emerging	350	C++
13	douban/paracel Distributed training framework with parameter server	42	Emerging	338	C++
14	jetsonhacks/installTensorFlowTX2 Install TensorFlow on the NVIDIA Jetson TX2 Development Kit	42	Emerging	166	Shell
15	modelbox-ai/modelbox A high performance, high expansion, easy to use framework for AI...	41	Emerging	160	C++
16	RedisAI/redis-inference-optimization A Redis module for serving tensors and executing deep learning graphs	41	Emerging	840	C
17	dmlc/tensorboard Standalone TensorBoard for visualizing in deep learning	39	Emerging	371	Python
18	lakshayg/tensorflow-build TensorFlow binaries supporting AVX, FMA, SSE	38	Emerging	237	—
19	BradPita/Win-AI-Wheelhouse 🪟 为 Windows AI 开发者提供预编译 wheel 文件的集中仓库 \| 自动抓取并整理 PyTorch、Flash...	37	Emerging	52	—
20	usc-isi/PipeEdge PipeEdge: Pipeline Parallelism for Large-Scale Model Inference on...	36	Emerging	38	Python
21	CINOAdam/nvml-unified-shim NVML unified memory shim for NVIDIA DGX Spark Grace Blackwell GB10 - enables...	35	Emerging	2	C
22	torrvision/crayon A language-agnostic interface to TensorBoard	35	Emerging	781	Python
23	j3soon/tbparse Load tensorboard event logs as pandas DataFrames for scientific plotting;...	35	Emerging	209	Python
24	Oabraham1/chronos Chronos is a time-based GPU partitioning utility that allows multiple users...	35	Emerging	26	C++
25	lakshayg/tensorflow-build-archived TensorFlow binaries supporting AVX, FMA, SSE	32	Emerging	1,902	Shell
26	OutofAi/tiny-cuda-nn-wheels This repository facilitates the creation of Python wheel files (.whl) from...	32	Emerging	12	—
27	KumaTea/pytorch-aarch64 PyTorch wheels (whl) & conda for aarch64 / ARMv8 / ARM64	32	Emerging	230	HTML
28	KumaTea/tensorflow-aarch64 TensorFlow wheels (whl) for aarch64 / ARMv8 / ARM64	32	Emerging	141	HTML
29	grc-iit/mega_mmap MegaMmap: Blurring the Boundary Between Memory and Storage for...	31	Emerging	4	C++
30	mind/wheels Performance-optimized wheels for TensorFlow (SSE, AVX, FMA, XLA, MPI)	31	Emerging	885	—
31	endernewton/c2board Tensorboard for Caffe2	31	Emerging	33	Python
32	zenetio/Install-VScode-On-TX2 How to install VScode on Nvidia TX2	31	Emerging	24	—
33	TGSAI/mdio-cpp C++, Cloud native, scalable storage engine for various types of energy data.	30	Emerging	6	C++
34	marnovo/macOS-eGPU-CUDA-guide Set up CUDA for machine learning (and gaming) on macOS using a NVIDIA eGPU	30	Emerging	229	Shell
35	elggem/tensorflow_node Tensorflow based ROS node for evaluating deep learning algorithms.	29	Experimental	11	Python
36	jedld/tensor_stream A ground-up and standalone reimplementation of TensorFlow for ruby. Comes...	29	Experimental	506	Ruby
37	deganza/Install-TensorFlow-on-Mac-M1-GPU Install TensorFlow in a few steps on Mac M1/M2 with GPU support and benefit...	27	Experimental	50	Jupyter Notebook
38	matthewlam721/octopus-parallel Octopus: Block-Level GPU Scheduling for Variable-Length Batches	26	Experimental	26	Python
39	diyor28/tf-aarch64 Tensorflow prebuilt wheel for Linux running inside Docker container on Apple Silicon	24	Experimental	3	Vue
40	jaysonle27/NVDAAL-Driver 🖥️ Enable powerful AI/ML capabilities with the NVDAAL compute driver for...	23	Experimental	1	C
41	David366AI/mini2gguf mini2gguf is an open-source toolkit that converts compact AI models...	23	Experimental	1	C++
42	alfinauzikri/ROCm-RX6600XT AMD ROCm Installation Guide on RX 6600 XT + TensorFlow and PyTorch	23	Experimental	81	—
43	francinesinglestranded789/cosim-gpu Simulate AMD MI300X GPU with QEMU and gem5 to run real ROCm/HIP workloads...	22	Experimental	—	Shell
44	Suryady50458/asystem-amem 🚀 Enhance NCCL communication with AMem. Offload and restore GPU memory...	22	Experimental	—	C++
45	ogulcanaydogan/dgx-spark-llm-stack Pre-built PyTorch wheels and build scripts for NVIDIA DGX Spark (GB10,...	22	Experimental	—	Python
46	golu2011/NVSmiBar Monitor remote NVIDIA GPUs on macOS via SSH with live updates on...	22	Experimental	—	TypeScript
47	gabrielmaialva33/NVDAAL-Driver Open-source compute driver for RTX 40 series GPUs on macOS - Pure AI/ML power	21	Experimental	2	C
48	ponder-lab/Hybridize-Functions-Refactoring Refactorings for optimizing imperative TensorFlow clients for greater efficiency.	17	Experimental	—	Java
49	osmszk/m1-mac-python-install 「M1 MacにPythonの環境構築してみた」動画の資料です	16	Experimental	17	—
50	lentesdaana2021/mlx-trm 🛠️ Simplify model training with MLX for Tiny Recursive Models, focusing on...	15	Experimental	1	Python
51	sxhxliang/TensorBuild Deep Learning Designer Library for Scientists	14	Experimental	7	JavaScript
52	romizone/gpu-cpu-tpu-simulator Interactive web simulator comparing GPU, CPU & TPU performance for AI...	14	Experimental	—	HTML
53	mikeroySoft/awesome-ROCm A curated list of awesome open source projects, tools, and resources for AMD...	14	Experimental	—	—
54	WobiWanKenobi/pytorch-amd-gpu Instructions on how to use PyTorch on AMD GPU with Linux	13	Experimental	6	Python
55	dstalzjohn/py-poetry-m1 This repository contains a tutorial for using python poetry and tensorflow...	12	Experimental	16	Makefile
56	Qengineering/TensorFlow-Lite-Raspberry-Pi_64-bit TensorFlow Lite installation wheels for Raspberry Pi 64 OS	12	Experimental	3	—

Comparisons in this category

tensorflow and tensorflow-on-raspberry-pi (100 vs 44) tensorflow and tensorflow-on-arm (100 vs 44) tensorflow and intel-extension-for-tensorflow (100 vs 42) tensorflow-on-raspberry-pi and tensorflow-on-arm (44 vs 44)