# Machine Learning
34 tools tagged
showing 34 of 34 tools
Ray
Distributed AI compute engine for scaling Python and ML workloads
Ray is an open-source distributed computing framework built for scaling AI and Python applications from a laptop to thousands of GPUs. It provides libraries for distributed training, hyperparameter tuning, model serving, reinforcement learning, and data processing under a single unified API. Ray's public site highlights OpenAI and other enterprise users. Maintained by Anyscale with Apache-2.0 open-source licensing.
LLaMA-Factory
Unified framework for fine-tuning 100+ large language models
LLaMA-Factory is an open-source toolkit providing a unified interface for fine-tuning over 100 LLMs and vision-language models. It supports SFT, RLHF with PPO and DPO, LoRA and QLoRA for memory-efficient training, and continuous pre-training. The LLaMA Board web UI enables no-code configuration, while CLI and YAML workflows serve advanced users. Integrates with Hugging Face, ModelScope, vLLM, and SGLang for model deployment.
Gradio
Build and share ML web apps in Python with a few lines of code
Gradio is a 42K+ star Apache-2.0 Python library for building interactive web interfaces around machine-learning functions and apps. It supports 40+ components for text, images, audio, video, 3D, chat, plots, JSON, and dataframes. Apps can be hosted free on Hugging Face Spaces, shared through public links, or extended with server-side rendering, streaming, API clients, and MCP server support.
Unsloth
2x faster LLM fine-tuning with 70% less VRAM on a single GPU
Unsloth is an open-source framework for fine-tuning large language models up to 2x faster while using 70% less VRAM. Built with custom Triton kernels, it supports 500+ model architectures including Llama 4, Qwen 3, and DeepSeek on consumer NVIDIA GPUs. Unsloth Studio adds a no-code web UI for dataset creation, training observability, model comparison, and GGUF export for Ollama and vLLM deployment.
Pinecone
Fully managed vector database built for AI applications at production scale.
Pinecone is a leading managed vector database designed for high-performance similarity search at scale. Purpose-built for AI applications including RAG, recommendation systems, and semantic search. Offers managed serverless infrastructure with automatic scaling, filtering, hybrid retrieval, and namespacing. No infrastructure management required.
Ragie
Fully managed RAG-as-a-Service platform for enterprise AI applications
Ragie is a managed retrieval-augmented generation platform that handles document ingestion, indexing, and retrieval so developers can build grounded AI applications without managing vector databases or chunking pipelines. It connects to Google Drive, Notion, Slack, Confluence, and other enterprise data sources with simple APIs for hybrid search and entity extraction.
Weaviate
Open-source vector database for AI-native applications and semantic search.
Weaviate is an open-source vector database purpose-built for AI applications. Supports vector, keyword, and hybrid search with built-in vectorization modules for OpenAI, Cohere, Hugging Face, and more. Used for RAG pipelines, semantic search, recommendation engines, and multimodal search. Written in Go for high performance.
Chroma
Open-source embedding database — the AI-native way to store and query embeddings.
Chroma is an open-source embedding database designed for simplicity and developer experience. Runs in-memory, as a Python library, or as a client-server deployment. Popular for prototyping RAG applications, local development, and lightweight vector search. Integrates natively with LangChain, LlamaIndex, and OpenAI.
Elasticsearch
Distributed search and analytics engine for all types of data.
Elasticsearch is the world's most popular open-source search and analytics engine, powering search experiences for companies like Wikipedia, GitHub, Netflix, and Uber. Built on Apache Lucene, it provides near-real-time search, structured and unstructured data analysis, and machine learning capabilities. Part of the Elastic Stack (ELK), it handles log analytics, application search, security analytics, and observability at scale. Supports vector search for AI/RAG applications.
MLflow
Open-source platform for the complete machine learning lifecycle.
MLflow is an open-source platform for managing the end-to-end machine learning lifecycle. Covers experiment tracking, model packaging, model registry, and deployment. Created by Databricks and now a Linux Foundation project. Integrates with TensorFlow, PyTorch, scikit-learn, Hugging Face, and all major ML frameworks.
DB-GPT
AI-native data application framework with SQL generation and agents
DB-GPT is an open-source AI-native data app framework combining SQL generation, database chat, RAG, and multi-agent orchestration for data-centric workflows. It supports natural language to SQL conversion, automated data analysis, and custom data app development. Integrates with MySQL, PostgreSQL, SQLite, and more. 19,000+ GitHub stars, MIT licensed. Positioned as an alternative to MindsDB for teams building AI-powered data applications and internal database tools.
AI Scientist v2
Autonomous scientific discovery via agentic tree search
AI Scientist v2 is Sakana AI's source-available system distributed under the AI Scientist Source Code License for fully autonomous scientific research using LLM-powered agentic tree search. It generates hypotheses, designs experiments, writes and executes code, analyzes results, and produces publishable manuscripts without human intervention. The system uses progressive exploration with backtracking to navigate the research space efficiently.
Agent Lightning
Microsoft's zero-code-change RL trainer for AI agents
Agent Lightning is Microsoft Research's open-source framework that makes AI agents trainable through reinforcement learning with virtually zero code changes. Supports RL, Automatic Prompt Optimization, and Supervised Fine-tuning across any agent framework including LangChain, OpenAI Agents SDK, AutoGen, and CrewAI. 14K+ GitHub stars, ranked among Microsoft's top 50 most-starred projects.
Amazon SageMaker
AWS's fully managed machine learning service for building, training, and deploying ML models.
Amazon SageMaker is AWS's comprehensive ML platform covering data labeling, notebook environments, model training, hyperparameter tuning, model hosting, and MLOps pipelines. Supports all major ML frameworks. Offers SageMaker Studio as an integrated IDE. Used by enterprises for production-scale ML workloads.
BitNet
Microsoft's framework for running 1-bit large language models on consumer CPUs
BitNet is Microsoft's official inference framework for 1-bit quantized large language models that enables running models with up to 100 billion parameters on standard consumer CPUs without requiring a GPU. By leveraging extreme quantization where weights use only 1.58 bits on average, BitNet achieves dramatic reductions in memory footprint and computational cost while maintaining competitive output quality for many practical use cases.
Cactus
On-device AI inference engine for mobile and wearable applications
Cactus is a YC-backed low-latency AI engine for mobile and wearable devices that runs LLMs, transcription, embedding, and TTS models locally. It achieves 16-20 tok/sec on older devices and 70+ tok/sec on flagships with ARM SIMD kernels optimized for Snapdragon, Apple, and MediaTek processors. Supports Qwen, Gemma, Llama, DeepSeek with Flutter, React Native, and Kotlin SDKs.
ChatDev
Multi-agent software company simulation for automated development
ChatDev simulates an entire virtual software company through multi-agent collaboration where LLM-powered roles including CEO, CTO, programmer, tester, and designer work together to produce complete software. With 32,000+ GitHub stars and a NeurIPS 2025 accepted paper, it offers a novel approach to automated software development through role-based agent orchestration.
Deep Lake
AI data runtime for multimodal datasets and vector search
Deep Lake is an open-source AI data runtime from Activeloop for storing, versioning, and querying multimodal data and embeddings. It fits teams building RAG, training, evaluation, or dataset-heavy agent workflows that need a bridge between vector search, structured metadata, and large image, text, audio, or video collections.
DeepSeek Coder
State-of-the-art open-source code language models
DeepSeek Coder is a family of open-source code language models trained from scratch on 2 trillion tokens of code and natural language data. Available in sizes from 1B to 33B parameters, these models support 80+ programming languages with 16K context windows and fill-in-the-blank capabilities. DeepSeek Coder outperforms CodeLlama-34B on HumanEval and MBPP benchmarks while being commercially licensable under MIT.
DeepSpeed
Deep learning optimization for distributed training
DeepSpeed is Microsoft's open-source deep learning optimization library that makes distributed training and inference easy, efficient, and effective. Its ZeRO optimizer eliminates memory redundancies across data-parallel processes, enabling training of models with trillions of parameters. DeepSpeed supports 3D parallelism combining data, pipeline, and tensor parallelism, along with mixed precision training, gradient checkpointing, and CPU/NVMe offloading for memory-constrained environments.
Fairlearn
Python toolkit for assessing and mitigating ML model fairness issues
Fairlearn is a Microsoft-backed open-source Python toolkit that helps developers assess and improve the fairness of machine learning models. It provides metrics for measuring disparity across groups defined by sensitive features, mitigation algorithms that reduce unfairness while maintaining model performance, and an interactive visualization dashboard for exploring fairness-accuracy trade-offs. Integrated with scikit-learn and Azure ML's Responsible AI dashboard.
FiftyOne
Open-source toolkit for curating datasets and evaluating visual AI models
FiftyOne is an open-source Python toolkit from Voxel51 for building high-quality datasets and better computer-vision and multimodal AI models. It pairs a browser-based visualization App with programmatic dataset curation, embeddings, similarity search, and model-evaluation workflows.
Jupyter
Interactive computing notebooks for data science
Jupyter is the open-source interactive computing platform providing notebook interfaces for data science, ML, scientific computing, and education. Notebooks combine live code, equations, visualizations, and narrative text. Supports 40+ languages via kernels including Python, R, Julia, and Scala. JupyterLab provides a modern IDE-like interface. JupyterHub enables multi-user deployments. The standard tool for computational research and data exploration worldwide.
Mage AI
Modern data pipeline orchestration with built-in AI
Mage AI is an open-source data pipeline orchestration tool positioned as a modern alternative to Apache Airflow. It provides a visual pipeline editor, native AI integrations for generating pipeline code, real-time streaming support, and built-in data quality checks. Mage handles batch and streaming workloads with a developer-friendly notebook-style interface and deploys to any cloud provider.
MediaPipe
On-device ML solutions for mobile and edge AI
MediaPipe is Google's open-source framework for building on-device machine learning pipelines across mobile, web, desktop, and edge platforms. It provides pre-built solutions for face detection, hand tracking, pose estimation, object detection, image classification, text classification, and on-device LLM inference. MediaPipe runs entirely locally without cloud dependencies, supporting Android, iOS, Python, and web browsers.
Memori
SQL-native memory infrastructure for AI agents and applications
Memori is an AI memory engine that provides persistent, queryable memory for agents and applications using SQL-native storage. It stores structured memories with semantic search, temporal awareness, and relationship tracking, enabling AI systems to remember user preferences, past interactions, and contextual facts across sessions. With 12,900 GitHub stars, it offers a database-native approach to the agent memory problem.
Notte
Browser automation framework turning websites into action APIs
Notte is a browser automation framework for AI agents that converts any website into a structured action API. Instead of scraping pages for text, Notte lets agents interact with sites — clicking buttons, filling forms, and navigating flows. Built with hybrid AI-plus-deterministic scripting, it includes digital personas, CAPTCHA solving, and proxy management for reliable automation at scale.
PaddleOCR
State-of-the-art OCR toolkit supporting 100+ languages from Baidu
PaddleOCR is an open-source OCR toolkit from Baidu's PaddlePaddle ecosystem with over 73,000 GitHub stars. It provides ultra-lightweight and high-accuracy text detection and recognition for 100+ languages including CJK, Arabic, and Indic scripts. The toolkit offers pre-trained models, easy deployment via pip, and server/edge inference options for document digitization workflows.
PrismML Bonsai
First commercially viable 1-bit LLMs that are 14x smaller and 8x faster
PrismML Bonsai delivers the first commercially viable 1-bit large language models with 8B, 4B, and 1.7B parameter variants. The 8B model runs in just 1GB of RAM versus 16GB for standard FP16 models, achieving 44 tokens per second on iPhone. Backed by $16.25M from Khosla Ventures and released under Apache 2.0, Bonsai makes capable LLMs practical for edge devices and resource-constrained environments.
SmoLAgents
Hugging Face's lightweight agent framework
smolagents is Hugging Face's lightweight agent framework for building AI agents that can use tools, write and execute code, and collaborate in multi-agent setups. Designed for simplicity with minimal abstractions — agents are just LLMs that write Python code to orchestrate tool calls rather than using JSON-based function calling. Supports any LLM provider, integrates with Hugging Face Hub for sharing tools and agents, and runs with as few as 1,000 lines of core library code.
Tecton
Enterprise feature platform for real-time ML
Tecton is an enterprise feature platform for building and serving ML features at scale. Created by the team behind Feast, it provides managed feature engineering, real-time feature computation from streaming data, feature monitoring, and a unified feature store with offline/online consistency. Used by production ML teams to eliminate training-serving skew and accelerate model deployment cycles.
TimesFM
Google's pretrained foundation model for zero-shot time-series forecasting
TimesFM is a pretrained time-series foundation model from Google Research that performs zero-shot forecasting on diverse datasets without task-specific training. It handles univariate and multivariate time series across domains including finance, logistics, energy, and infrastructure monitoring with accuracy competitive against traditional statistical methods like ARIMA and Prophet.
Whisper
OpenAI's open-source speech recognition model for any language
Whisper is OpenAI's open-source automatic speech recognition model trained on 680,000 hours of multilingual audio data. It supports transcription and translation across 99 languages with robust handling of accents, background noise, and technical vocabulary. Available in multiple model sizes from tiny (39M) to large (1.5B parameters) for balancing accuracy and speed.
verl
Production-grade reinforcement learning framework for LLM training
verl is an open-source reinforcement learning framework designed specifically for training and aligning large language models. Built for production use with support for distributed training across multiple GPUs and nodes, it implements RLHF, DPO, and other alignment algorithms that make LLMs follow instructions, avoid harmful outputs, and generate higher quality responses. Over 580 contributors and 20,000 GitHub stars signal strong adoption.