26 projects
TensorFlow
TensorFlow is an open-source machine learning framework developed by Google that enables numerical computation and large-scale machine learning. It provides a flexible system for defining and executing computations involving tensors, which are multi-dimensional arrays. The framework supports deep learning and neural networks across multiple platforms and devices.
47,170
6,145
$198M
PyTorch
PyTorch is an open source machine learning framework that accelerates the path from research prototyping to production deployment.
40,722
5,567
$123M
Keras
Keras is an open-source deep learning framework that provides a Python interface for artificial neural networks. It serves as a high-level API capable of running on top of multiple backend engines including TensorFlow, offering a user-friendly, modular, and extensible way to build and train deep learning models.
17,970
2,614
$7.8M
PaddlePaddle
PaddlePaddle is an open-source deep learning platform developed by Baidu that provides a comprehensive suite of tools for AI model development, training, and deployment. It features an easy-to-use API, high performance distributed training capabilities, and extensive support for various deep learning applications including computer vision, natural language processing, and speech recognition.
9,224
478
$107M
PyTorch Lightning
PyTorch Lightning is a lightweight PyTorch wrapper that helps researchers and engineers train deep learning models with high performance at scale. It provides a high-level interface for organizing PyTorch code, automating complex training features like distributed training, mixed precision, and model checkpointing while removing boilerplate code.
7,410
1,477
$4.5M
JAX
JAX is a high-performance numerical computing and machine learning library that combines Numpy's familiar API with GPU and TPU hardware acceleration. It features automatic differentiation, just-in-time compilation, and enables writing transformable numerical programs.
6,107
1,124
$20M
DeepRec
The mission of the Project is to develop a high-performance recommendation deep learning framework.
2,724
276
$160M
fastai
fastai is a deep learning library and educational platform that simplifies training fast and accurate neural networks using modern best practices. It provides high-level components for training models with state-of-the-art results in computer vision, natural language processing, tabular data analysis, and collaborative filtering.
2,601
544
$5.5M
Horovod
Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
2,264
335
$2.8M
tinygrad
tinygrad is a minimalist deep learning framework designed to be simple, lightweight, and easy to understand. It implements neural network operations and autograd functionality in pure Python, allowing users to build and train neural networks with minimal dependencies.
1,921
338
$13M
MindSpore
MindSpore is an open-source deep learning framework that provides a unified training and inference experience across different devices and platforms. It features automatic differentiation, dynamic debugging capabilities, and hardware optimization for AI model development.
1,726
71
$122M
TT-Metal
TT-Metal is a software framework for programming Tenstorrent AI accelerator chips, providing low-level hardware access and control capabilities for machine learning workloads
1,346
76
$67M
FlagAI
The mission of the Project is to develop a fast, easy-to-use and extensible toolkit for large-scale AI modeling, with the goal of supporting training, fine-tuning, and deployment of large-scale models on various downstream tasks with multi-modality.
364
32
$47M
Elastic Deep Learning (EDL)
Elastic Deep Learning using PaddlePaddle and Kubernetes.
56
9
$692K
Caffe
Caffe: a fast open framework for deep learning.
Chainer
A flexible framework of neural networks for deep learning
Colossal-AI
Making large AI models cheaper, faster and more accessible
MLX
MLX: An array framework for Apple silicon
MNN
MNN is a blazing fast, lightweight deep learning framework, battle-tested by business-critical use cases in Alibaba. Full multimodal LLM Android App:[MNN-LLM-Android](./apps/Android/MnnLlmChat/README.md)
MXNet
Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more
NeMo
A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)
OneFlow
OneFlow is a deep learning framework designed to be user-friendly, scalable and efficient.
Relax
Relax is a Linux Foundation project focused on providing tools or frameworks for system relaxation, potentially offering solutions for stress management, system optimization, or resource allocation in computing environments.
TensorFlow.js
A WebGL accelerated JavaScript library for training and deploying ML models.