AI Open Source Frameworks

Discover and learn various AI and machine learning frameworks to help you quickly build, train, and deploy AI models

Found 30 frameworks
PyTorch

Deep learning framework developed by Facebook, known for its dynamic computation graph and Python-first design philosophy, popular among researchers

General FrameworksDeep LearningDynamic GraphPythonGeneral Framework73.0k
TensorFlow

End-to-end open-source machine learning platform developed by Google, supporting multiple languages and platforms with strong production deployment capabilities

General FrameworksDeep LearningStatic GraphMulti-platformEnterprise-grade177.0k
Hugging Face Transformers

Most popular NLP library providing thousands of pre-trained models and easy-to-use APIs, simplifying the use of Transformer models

NLP FrameworksNLPTransformerPre-trained ModelsTransfer Learning115.0k
JAX

High-performance numerical computing library developed by Google, combining NumPy and automatic differentiation, designed for high-performance machine learning research

General FrameworksHigh-performance ComputingAutomatic DifferentiationGPU/TPUFunctional25.0k
Keras

User-friendly deep learning API focused on fast experimentation, can be used with TensorFlow, CNTK, and Theano

General FrameworksDeep LearningHigh-level APIUser-friendlyRapid Prototyping59.0k
Detectron2

Computer vision library developed by Facebook AI, implementing state-of-the-art object detection and segmentation algorithms

Computer VisionComputer VisionObject DetectionImage SegmentationPyTorch26.0k
ONNX

Open Neural Network Exchange format allowing AI developers to transfer models between different frameworks, promoting framework interoperability

Model DeploymentModel ExchangeCross-frameworkDeploymentStandardization15.0k
Deep Graph Library (DGL)

Framework specifically designed for graph neural networks, supporting PyTorch, TensorFlow, and MXNet backends

Graph Neural NetworksGraph Neural NetworksGNNKnowledge GraphsRelational Data12.0k
spaCy

Industrial-strength natural language processing library focused on efficient text processing and NLP tasks in production environments

NLP FrameworksNLPText ProcessingNamed Entity RecognitionDependency Parsing27.0k