Keras alternatives for AI agents.

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

TensorFlow-based neural network library

JAX-based neural network library

A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility

A list of popular github projects related to deep learning

Deep learning library featuring a higher-level API for TensorFlow.

A python library for user-friendly forecasting and anomaly detection on time series.

Probabilistic reasoning and statistical analysis in TensorFlow

Deep universal probabilistic programming with Python and PyTorch

Core ML tools contain supporting tools for Core ML model conversion, editing, and validation.

Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

Supercharge Your Model Training

Probabilistic time series modeling in Python

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

Machine Learning From Scratch. Bare bones NumPy implementations of machine learning models and algorithms with a focus on accessibility. Aims to cover everything from linear regression to deep learning.