domain: machine-learning
3991 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| omry/omegaconf OmegaConf is a hierarchical Python configuration library that merges configs from YAML files, dataclasses/objects, and CLI arguments behind… | 84 | 2426 | stable |
| tflearn/tflearn TFLearn is a modular deep learning library providing a higher-level, Keras-like API on top of TensorFlow for building and training neural n… | 23 | 9576 | maintenance |
| google/tunix Tunix is a lightweight JAX-based library for post-training large language models, supporting supervised fine-tuning, preference optimizatio… | 83 | 2415 | active |
| AI-Hypercomputer/maxtext MaxText is a high-performance, scalable open-source LLM training library written in pure Python/JAX, targeting Google Cloud TPUs and GPUs. … | 97 | 2408 | active |
| resemble-ai/resemble-enhance Resemble Enhance is an AI-powered Python tool that improves speech quality through denoising and enhancement, using a denoiser module and a… | 17 | 2397 | active |
| sjwhitworth/golearn GoLearn is a 'batteries included' machine learning library for Go with a scikit-learn-style Fit/Predict interface. It provides data loading… | 32 | 9439 | maintenance |
| nv-tlabs/3dgrut NVIDIA's official implementations of 3D Gaussian Ray Tracing (3DGRT) and 3D Gaussian Unscented Transform (3DGUT), which render volumetric G… | 71 | 2390 | active |
| AminHP/gym-anytrading gym-anytrading is a collection of OpenAI Gym environments for developing and testing reinforcement learning-based trading algorithms in FOR… | 23 | 2386 | stable |
| ailia-ai/ailia-models A collection of 400+ pre-trained state-of-the-art AI models (object detection, pose estimation, speech recognition, LLMs, image generation,… | 77 | 2385 | active |
| google/neural-tangents Neural Tangents is a Python library built on JAX for defining, training, and evaluating neural networks of both finite and infinite width. … | 10 | 2383 | stable |
| microsoft/Olive Olive is Microsoft's AI model optimization toolkit for the ONNX Runtime, automating finetuning, conversion, quantization, and compression o… | 91 | 2382 | active |
| lupinemachines/lupine LUPINE is a GPU-over-IP bridge that lets machines without local GPUs attach NVIDIA GPUs running on remote servers over the network. It expo… | 89 | 2382 | active |
| FMInference/FlexLLMGen FlexLLMGen is a high-throughput generation engine for running large language models on a single GPU with limited memory. It uses IO-efficie… | 10 | 9352 | maintenance |
| apple/axlearn AXLearn is a Python deep learning library built on JAX and XLA for developing and training large-scale models, with an object-oriented conf… | 71 | 2372 | active |
| zai-org/GLM-V GLM-V is the open-source repository for Zhipu AI's GLM-4.6V, GLM-4.5V, and GLM-4.1V-Thinking vision-language models, which perform versatil… | 60 | 2370 | active |
| jik876/hifi-gan The official PyTorch implementation of HiFi-GAN, a generative adversarial network that converts mel-spectrograms into high-fidelity 22.05 k… | 32 | 2367 | stable |
| uptrain-ai/uptrain UpTrain is an open-source Python platform for evaluating and improving LLM applications, offering 20+ preconfigured evaluation checks cover… | 23 | 2362 | active |
| alibaba/EasyRec EasyRec is a TensorFlow-based framework from Alibaba for building large-scale deep learning recommendation models covering matching, rankin… | 65 | 2356 | active |
| facebookresearch/perception_models Meta's Perception Models repository hosting state-of-the-art image, video, and audio encoders (Perception Encoder, PE) and a multimodal lan… | 54 | 2353 | active |
| mlech26l/ncps A Python package providing PyTorch and TensorFlow/Keras implementations of Neural Circuit Policies (NCPs), including liquid time-constant (… | 23 | 2345 | active |
| ComflowySpace ComflowySpace is an open-source desktop application that wraps ComfyUI/Stable Diffusion into a friendlier, app-like interface for generatin… | 17 | 2345 | active |
| HybridRobotics/whole_body_tracking BeyondMimic's official motion tracking training code, a humanoid control framework built on Isaac Lab that trains sim-to-real-ready whole-b… | 69 | 2344 | active |
| maitrix-org/llm-reasoners LLM Reasoners is a Python library that provides a standardized modular framework for advanced LLM reasoning, implementing algorithms like M… | 31 | 2343 | active |
| NVlabs/ProtoMotions ProtoMotions is a GPU-accelerated simulation and reinforcement learning framework for training physically simulated digital humans and huma… | 66 | 2338 | active |
| tracel-ai/cubecl CubeCL is a Rust language extension, JIT compiler, and runtime set for writing high-performance GPU compute kernels in plain Rust. A single… | 81 | 2334 | active |
| refuel-ai/autolabel Autolabel is a Python library for labeling, cleaning, and enriching text datasets using large language models like GPT-4, Claude, and open-… | 25 | 2329 | active |
| ufoym/imbalanced-dataset-sampler A PyTorch dataset sampler (ImbalancedDatasetSampler, pip package torchsampler) that rebalances class distributions during training by overs… | 64 | 2326 | stable |
| google-deepmind/optax Optax is a gradient processing and optimization library for JAX, offering composable building blocks like optimizers and loss functions. It… | 82 | 2325 | stable |
| labmlai/labml A Python library for tracking and monitoring deep learning experiments, with a self-hostable server app for viewing metrics and hardware us… | 30 | 2325 | active |
| facebookresearch/schedule_free A PyTorch library implementing schedule-free optimizers (SGD, AdamW, RAdam variants) that remove the need for learning rate schedules or sp… | 67 | 2323 | active |
| vearch/vearch Vearch is a cloud-native distributed vector database for efficient similarity search of embedding vectors, supporting hybrid vector and sca… | 82 | 2321 | active |
| Cadene/pretrained-models.pytorch A Python library providing pretrained ConvNet models (ResNet, ResNeXt, InceptionV4, Xception, NASNet, SENet, DPN, etc.) for PyTorch behind … | 32 | 9099 | maintenance |
| iver56/audiomentations Audiomentations is a Python library for audio data augmentation with an API inspired by albumentations. It provides fast CPU-based waveform… | 67 | 2314 | stable |
| facebookresearch/ImageBind A PyTorch library from Meta AI implementing ImageBind, a model that learns a joint embedding space across six modalities: images, text, aud… | 54 | 9064 | maintenance |
| ALOHA ALOHA is an open-source hardware and software system for low-cost bimanual robot teleoperation and demonstration data collection, built on … | 31 | 2310 | active |
| IDEA-Research/detrex detrex is an open-source PyTorch-based research platform and toolbox for DETR-style Transformer detection algorithms, built on top of Detec… | 41 | 2306 | active |
| claimed-framework/claimed CLAIMED (C3) is a component compiler that turns Jupyter notebooks, Python scripts, and R scripts into portable containerized AI components.… | 88 | 2305 | active |
| apache/mahout Apache Mahout is an Apache Software Foundation project providing Qumat, a high-level Python library for quantum computing with a unified AP… | 86 | 2304 | active |
| frgfm/torch-cam TorchCAM is a Python library that extracts class activation maps (CAMs) from PyTorch CNN classifiers, supporting many CAM variants such as … | 72 | 2304 | active |
| YvanYin/Metric3D Metric3D is the official PyTorch implementation of Metric3Dv1 and Metric3Dv2, monocular geometric foundation models that predict metric dep… | 33 | 2302 | active |
| XiaomiMiMo/MiMo Xiaomi's MiMo is a 7B-parameter reasoning language model trained from pretraining through posttraining with reinforcement learning, release… | 30 | 2299 | active |
| NVlabs/nvdiffrec nvdiffrec is NVIDIA's official implementation of a CVPR 2022 oral paper that jointly optimizes triangular 3D meshes, PBR materials, and lig… | 76 | 2296 | stable |
| emgucv/emgucv Emgu CV is a cross-platform .NET wrapper for the OpenCV image processing library, allowing OpenCV functions to be called from .NET-compatib… | 74 | 2294 | active |
| OpenHelix-Team/VLA-Adapter VLA-Adapter is the official implementation of a tiny-scale vision-language-action model that bridges vision-language representations to rob… | 50 | 2294 | active |
| mflux-community/mflux MFLUX is a native MLX implementation of state-of-the-art generative image and video models (Flux, Qwen-Image, Z-Image, and others), ported … | 90 | 2291 | active |
| huggingface/picotron Picotron is a minimalist, hackable distributed training framework for pre-training Llama-like large language models using 4D parallelism (d… | 40 | 2289 | active |
| benedekrozemberczki/karateclub Karate Club is a Python library of unsupervised machine learning algorithms for graph-structured data, built as an extension of NetworkX. I… | 23 | 2286 | stable |
| sqliteai/warp WARP is a dependency-free, embeddable C inference engine that runs trillion-parameter mixture-of-experts models like Kimi K3 by streaming a… | 80 | 2284 | active |
| codeplea/genann Genann is a minimal, well-tested C99 library for training and running feedforward artificial neural networks. It is contained in a single s… | 97 | 2280 | stable |
| fairlearn/fairlearn Fairlearn is a Python package for assessing and mitigating fairness issues in machine learning models. It provides fairness metrics for eva… | 84 | 2277 | active |
| VexDB-THU/VexDB-Lite VexDB-Lite is a cross-platform vector similarity search engine implemented as plugins for PostgreSQL, DuckDB, and SQLite, sharing a graph i… | 80 | 2275 | active |
| feature-engine/feature_engine Feature-engine is a Python library for feature engineering and feature selection with transformers that follow scikit-learn's fit/transform… | 86 | 2272 | stable |
| Liuziyu77/Visual-RFT Official research code for Visual-RFT and Visual-ARFT, applying GRPO-based reinforcement fine-tuning with rule-based verifiable rewards to … | 42 | 2271 | active |
| OlafenwaMoses/ImageAI ImageAI is a Python library that lets developers add computer vision capabilities like image classification, object detection, and video ob… | 23 | 8877 | maintenance |
| OpenDCAI/DataFlex DataFlex is a data-centric training framework built on LLaMA-Factory that dynamically schedules LLM training data during optimization. It p… | 67 | 2270 | active |
| 666DZY666/micronet micronet is a Python library for deep neural network model compression and deployment built on PyTorch. It provides quantization (QAT, PTQ,… | 41 | 2266 | active |
| ermig1979/Simd Simd Library is a free open-source C++ image processing and machine learning library with a C API and Python wrapper. Its algorithms are ha… | 98 | 2265 | active |
| yemount/pose-animator Pose Animator is a browser-based tool that animates 2D SVG vector characters in real time using pose and face keypoints detected by PoseNet… | 32 | 8852 | maintenance |
| hkchengrex/MMAudio MMAudio is a PyTorch-based model for generating synchronized audio from video and/or text inputs, using multimodal joint training across au… | 43 | 2264 | active |
| radixark/miles Miles is an open-source, enterprise-grade reinforcement learning framework for large-scale LLM and VLM post-training, forked from and co-ev… | 72 | 2263 | active |
| aws/sagemaker-python-sdk The SageMaker Python SDK is an open-source Python library for training and deploying machine learning models on Amazon SageMaker. It suppor… | 95 | 2261 | active |
| opendatalab/DocLayout-YOLO DocLayout-YOLO is a real-time YOLO-v10-based model for detecting document layout elements (text blocks, tables, figures, etc.) in diverse d… | 29 | 2258 | active |
| RosettaCommons/RoseTTAFold RoseTTAFold is the official implementation of a deep learning system for predicting protein structures and interactions using a three-track… | 23 | 2258 | stable |
| skfolio/skfolio skfolio is a Python library for portfolio optimization, factor model construction, and risk management built on top of scikit-learn. It pro… | 93 | 2256 | active |
| langfengQ/verl-agent verl-agent is an extension of the veRL framework for training LLM and VLM agents via reinforcement learning, featuring step-independent mul… | 56 | 2256 | active |
| LAION-AI/CLAP CLAP (Contrastive Language-Audio Pretraining) is a Python library from LAION that learns joint audio and text embeddings using contrastive … | 42 | 2256 | stable |
| PrismML-Eng/Bonsai-demo A demo repository for running PrismML's Bonsai family of 1-bit and ternary-quantized language models locally via llama.cpp and MLX. It prov… | 59 | 2252 | active |
| CoinCheung/pytorch-loss A PyTorch library providing a collection of loss functions (focal loss, triplet loss, AMSoftmax, label-smooth CE, dice loss, lovasz-softmax… | 32 | 2252 | active |
| TuringLang/Turing.jl Turing.jl is a probabilistic programming library in Julia for Bayesian inference. Models are defined with an @model macro resembling mathem… | 95 | 2251 | active |
| facebookresearch/fvcore fvcore is a lightweight Python core library providing common functionality shared across FAIR's computer vision frameworks such as Detectro… | 76 | 2250 | stable |
| azavea/raster-vision Raster Vision is an open source Python library and low-code framework for building computer vision models on satellite, aerial, and other l… | 61 | 2240 | active |
| Accenture/AmpliGraph AmpliGraph is a Python library built on TensorFlow 2 for representation learning on knowledge graphs. It provides neural models (TransE, Di… | 84 | 2238 | active |
| facebookresearch/fairchem FAIR Chemistry's centralized Python library of machine learning models, datasets, and applications for materials science and quantum chemis… | 99 | 2231 | active |
| NVIDIA/vid2vid A PyTorch implementation of NVIDIA's video-to-video synthesis method for generating high-resolution (e.g., 2048x1024) photorealistic videos… | 32 | 8692 | maintenance |
| facebookresearch/DiT Official PyTorch implementation of Diffusion Transformers (DiT) from the paper 'Scalable Diffusion Models with Transformers', including mod… | 10 | 8689 | maintenance |
| tensorchord/envd envd is a command-line tool that creates reproducible, container-based development environments for AI/ML work using a simple Python-like b… | 86 | 2227 | active |
| rapidsai/cugraph cuGraph is NVIDIA's RAPIDS collection of GPU-accelerated graph analytics libraries, offering Python, C, and C++ APIs for building graphs an… | 94 | 2225 | active |
| ellisdg/3DUnetCNN A PyTorch library for building, training, and applying 3D U-Net convolutional neural networks for medical image segmentation. It provides c… | 45 | 2224 | active |
| floneum/kalosm Kalosm is a Rust ecosystem of crates providing simple interfaces for running pre-trained language, audio, and image models locally or remot… | 64 | 2223 | active |
| dstackai/dstack dstack is an open-source, vendor-agnostic control plane for GPU provisioning and orchestration that works across GPU clouds, Kubernetes, an… | 95 | 2221 | active |
| ahmedfgad/GeneticAlgorithmPython PyGAD is an open-source Python 3 library for implementing the genetic algorithm to optimize single- and multi-objective problems. It can al… | 80 | 2220 | active |
| DerwenAI/pytextrank PyTextRank is a Python implementation of TextRank and related textgraph algorithms (PositionRank, Biased TextRank, TopicRank) as a spaCy pi… | 63 | 2218 | active |
| lifeiteng/vall-e An unofficial PyTorch implementation of VALL-E, a zero-shot text-to-speech model that treats TTS as a conditional language modeling task ov… | 40 | 2215 | active |
| unrealcv/unrealcv UnrealCV is an open-source Unreal Engine plugin that connects computer vision research to virtual worlds by exposing a command API and Pyth… | 67 | 2209 | active |
| MVIG-SJTU/AlphaPose AlphaPose is an open-source real-time multi-person full-body pose estimation and tracking system built on PyTorch. It detects human keypoin… | 32 | 8596 | maintenance |
| DigitalPhonetics/IMS-Toucan IMS Toucan is a PyTorch-based toolkit for training and running state-of-the-art, controllable text-to-speech synthesis, home of the massive… | 63 | 2207 | active |
| RubixML/ML Rubix ML is a high-level open-source machine learning and deep learning library for PHP, offering 40+ supervised and unsupervised algorithm… | 99 | 2204 | active |
| NovaSky-AI/SkyRL SkyRL is a modular full-stack reinforcement learning library for post-training large language models, combining a training framework (skyrl… | 80 | 2201 | active |
| thuml/iTransformer Official PyTorch implementation of iTransformer, an ICLR 2024 Spotlight paper that inverts the Transformer architecture for multivariate ti… | 41 | 2200 | stable |
| lucidrains/lion-pytorch A PyTorch implementation of the Lion optimizer (Evolved Sign Momentum), discovered by Google Brain via genetic algorithms and claimed to ou… | 62 | 2199 | active |
| kubeflow/trainer Kubeflow Trainer is a Kubernetes-native platform for distributed AI model training and LLM fine-tuning across frameworks like PyTorch, JAX,… | 95 | 2198 | active |
| NX-AI/xlstm Official PyTorch implementation of xLSTM, an extended Long Short-Term Memory recurrent architecture with exponential gating and matrix memo… | 59 | 2198 | active |
| tensorflow/tfx TensorFlow Extended (TFX) is an end-to-end, Google-production-scale platform for building and deploying production machine learning pipelin… | 91 | 2190 | stable |
| MinishLab/model2vec Model2Vec is a Python library that distills any sentence transformer into a tiny, fast static embedding model by computing one fixed vector… | 90 | 2186 | active |
| NVIDIA-NeMo/DataDesigner NeMo Data Designer is an open-source Python orchestration framework from NVIDIA for generating high-quality synthetic datasets from scratch… | 83 | 2185 | active |
| levy-street/world-of-claudecraft World of ClaudeCraft is a free, open-source classic-style browser MMO built in TypeScript with Three.js, playable online with persistent ch… | 80 | 2185 | active |
| tensorflow/quantum TensorFlow Quantum is a Python framework for hybrid quantum-classical machine learning that integrates Cirq quantum circuits with TensorFlo… | 84 | 2180 | active |
| osqp/osqp OSQP is a C library for solving convex quadratic programs using an ADMM-based operator splitting method, with interfaces to Python, Julia, … | 51 | 2177 | stable |
| Harry24k/adversarial-attacks-pytorch Torchattacks is a PyTorch library providing implementations of adversarial attacks to generate adversarial examples against deep learning m… | 23 | 2177 | active |
| intel/intel-extension-for-transformers Intel's toolkit for accelerating transformer-based GenAI/LLM workloads on Intel platforms, offering state-of-the-art compression (e.g., INT… | 10 | 2174 | active |