ML and data science for projects built with AI agents
NumPy, pandas, PyTorch, scikit-learn. 28 tools ranked by how many of 12,697 public projects built with AI coding agents use them, with downloads and GitHub stars. Updated Oct 10, 2026.
| Tool | Used in agent projects | Downloads / wk | GitHub stars | License |
|---|---|---|---|---|
| NumPy pypi Fundamental Python library for numerical computing with arrays and matrices. |
487 (3.8%) | 21M | 33K | — |
| pandas pypi Data manipulation and analysis library with DataFrames for structured data. |
383 (3.0%) | 143M | 50K | — |
| scikit-learn pypi Python machine learning library with classification, regression and clustering algorithms. |
160 (1.3%) | 44.3M | 67.5K | — |
| SciPy pypi Fundamental algorithms for scientific computing with optimized numerical routines. |
141 (1.1%) | 72.2M | 15.1K | — |
| PyTorch pypi Deep learning framework with tensors and dynamic neural networks with GPU support. |
109 (0.9%) | 13.2M | 104K | — |
| Sentence Transformers pypi Pre-trained models for semantic embeddings, retrieval and reranking. |
83 (0.7%) | 5.1M | 19.2K | — |
| Hugging Face Transformers pypi State-of-the-art NLP and multimodal models for inference and training. |
80 (0.6%) | 24.8M | 167K | Apache 2.0 License |
| NetworkX pypi Python package for creating, analyzing and manipulating complex networks and graphs. |
73 (0.6%) | 50.2M | 17.3K | — |
| joblib pypi Lightweight pipelining and parallel computing for Python functions with caching support. |
65 (0.5%) | 38.7M | 4,403 | — |
| PyArrow pypi Python library implementing Apache Arrow for efficient columnar data storage and processing. |
60 (0.5%) | 92.6M | 17.2K | — |
| IPykernel pypi Jupyter kernel for interactive Python computing and data science notebooks. |
57 (0.4%) | 13.2M | 733 | — |
| Chroma pypi Vector database for storing and querying embeddings in AI/ML applications. |
56 (0.4%) | 1.7M | 29.5K | — |
| torchvision pypi Image and video datasets and pre-trained models for PyTorch deep learning. |
44 (0.3%) | 3.7M | 17.9K | BSD |
| FAISS pypi Library for efficient similarity search and clustering of dense vectors at scale. |
42 (0.3%) | 2.9M | 41.1K | — |
| jiter pypi High-performance JSON parser for Python with iterable streaming support. |
36 (0.3%) | 80.8M | 555 | — |
| yfinance pypi Download historical market data from Yahoo! Finance API for analysis. |
36 (0.3%) | 4M | 25.5K | Apache-2.0 |
| XGBoost pypi Gradient boosting library for fast, accurate machine learning on structured data. |
30 (0.2%) | 7.1M | 28.8K | — |
| Jupyter pypi Interactive notebook environment for data analysis, visualization and code experimentation. |
28 (0.2%) | 3.2M | — | BSD |
| spaCy pypi Industrial-strength NLP library for text processing, tokenization and entity recognition. |
28 (0.2%) | 4.9M | 34K | MIT |
| SymPy pypi Computer algebra system in Python for symbolic mathematics and equation solving. |
28 (0.2%) | 30.8M | 15K | BSD |
| NLTK pypi Natural language processing toolkit with text analysis and NLP algorithms. |
25 (0.2%) | 10.1M | 14.7K | Apache License, Version 2.0 |
| statsmodels pypi Comprehensive statistical modeling and analysis library for Python. |
24 (0.2%) | 8.3M | 11.7K | — |
| mpmath pypi Python library for arbitrary-precision floating-point arithmetic. |
23 (0.2%) | 30.4M | 1,220 | — |
| ONNX Runtime pypi Fast inference runtime for machine learning models in ONNX format. |
23 (0.2%) | 20.8M | 22K | MIT License |
| Hugging Face Datasets pypi Community-driven library providing easy access to thousands of ML datasets. |
22 (0.2%) | 15.6M | 22K | Apache 2.0 |
| safetensors pypi Safe and fast serialization format for machine learning model tensors used by Hugging Face and PyTorch. |
22 (0.2%) | 19.7M | 3,915 | — |
| Gradio pypi Python library to quickly build interactive web UIs for machine learning models. |
20 (0.2%) | 1.3M | 43.7K | — |
| Optuna pypi Python framework for automated hyperparameter optimization and neural architecture search. |
20 (0.2%) | 3.1M | 14.9K | — |
Nothing here matches. Search all tools →
Other categories
Web frameworks UI components and icons Styling and CSS Animation and motion 3D and graphics Charts and data visualisation Forms, validation and schemas State and data fetching Markdown, editors and content Desktop and mobile apps Utilities Model SDKs Agent, MCP and tool libraries Prompts, structured output and evals Databases, ORMs and clients SDKs for hosted services Auth and security libraries Jobs, queues and workflows Files, images and media Scraping, browsers and parsing HTTP servers and APIs HTTP clients and networking Testing Linting and formatting Languages, types and runtimes Build tools and bundlers Dev workflow and config Logging, monitoring and analytics Crypto and web3 Game development