Machine learning libraries, mapped by language and task

1,170 libraries from the Awesome Machine Learning list, each with its GitHub stars and last commit, so the maintained ones stand out from the abandoned.

Bubble size

Browse by task

updated this yearquiet for over a yearquiet for 3+ years

General-purpose ML609 librariesFrameworks and toolkits for training models of any kind, from scikit-learn to PyTorch

Deep learning251 librariesNeural network frameworks, model zoos and training tools

LLMs and agents58 librariesLanguage model frameworks, vector databases, RAG and LLM evaluation

Computer vision72 librariesImage processing, detection, segmentation and face recognition

Natural language processing144 librariesTokenizers, parsers, embeddings, text classification and NLP toolkits

Speech and audio10 librariesSpeech recognition toolkits and on-device speech runtimes

Reinforcement learning31 librariesEnvironments, agents and training libraries for reinforcement learning

Data analysis and visualization202 librariesData frames, numerical computing, statistics and charting

MLOps and tooling98 librariesExperiment tracking, model serving, feature stores, AutoML and vector databases

Math and statistics20 librariesNumerical, statistical and geometry libraries plus language interop

Notebooks and examples91 librariesJupyter notebooks, book code, tutorials and Kaggle solutions

Learning resources12 librariesCurated sub-lists and interactive tools for learning machine learning

Browse by language