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 58 platform-optimized wheels frameworks tracked. 1 score above 70 (verified tier). The highest-rated is tensorflow/tensorflow at 87/100 with 194,185 stars. 3 of the top 10 are actively maintained.

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

Comparisons in this category

tensorflow and tensorflow-on-raspberry-pi (87 vs 51) tensorflow and tensorflow-on-arm (87 vs 51) tensorflow and intel-extension-for-tensorflow (87 vs 49) tensorflow-on-arm and tensorflow-on-raspberry-pi (51 vs 51)