Repository Index

An Open Source Machine Learning Framework for Everyone

• tensorflow
• machine-learning
• python
• deep-learning

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

• nlp
• natural-language-processing
• pytorch
• pytorch-transformers

Tensors and Dynamic neural networks in Python with strong GPU acceleration

• neural-network
• autograd
• gpu
• numpy

Deep Learning for humans

• deep-learning
• tensorflow
• neural-networks
• machine-learning

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

• jax

Caffe: a fast open framework for deep learning.

• deep-learning
• machine-learning
• vision

You like pytorch? You like micrograd? You love tinygrad! ❤️

The Modular Platform (includes MAX & Mojo)

• ai
• language
• machine-learning
• mojo

MLX: An array framework for Apple silicon

• mlx

The fastai deep learning library

• deep-learning
• machine-learning
• pytorch
• python

Convolutional Neural Networks

TranslatedPArallel Distributed Deep LEarning: Machine Learning Framework from Industrial Practice (Core framework of 'PaddlePaddle', high-performance single-machine and distributed training and cross-platform deployment for deep learning & machine learning)

• paddlepaddle
• deep-learning
• scalability
• machine-learning

Graph Neural Network Library for PyTorch

• pytorch
• geometric-deep-learning
• graph-neural-networks
• deep-learning

Minimalist ML framework for Rust

A WebGL accelerated JavaScript library for training and deploying ML models.

• deep-learning
• typescript
• webgl
• javascript

A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API

MNN: A blazing-fast, lightweight inference engine battle-tested by Alibaba, powering high-performance on-device LLMs and Edge AI.

• mnn
• convolution
• vulkan
• embedded-devices

Machine learning, in numpy

• machine-learning
• neural-networks
• topic-modeling
• gaussian-mixture-models

Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.

• autodiff
• deep-learning
• machine-learning
• rust

Tensor library for machine learning

• machine-learning

🤖 GPU accelerated Neural networks in JavaScript for Browsers and Node.js

• neural-network
• brain
• recurrent-neural-networks
• easy-to-use

Python package built to ease deep learning on graph, on top of existing DL frameworks.

• deep-learning
• graph-neural-networks

Suite of tools for deploying and training deep learning models using the JVM. Highlights include model import for keras, tensorflow, and onnx/pytorch, a modular and tiny c++ library for running math code and a java based math library on top of the core c++ library. Also includes samediff: a pytorch/tensorflow like library for running deep learn...

• java
• gpu
• deeplearning
• neural-nets

http://torch.ch