Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
12 similar packages · ranked by shared classifiers & health score · View horovod →
The Deep Learning framework to train, deploy, and ship AI products Lightning fast.
Open Source Differentiable Computer Vision Library for PyTorch
PMB (Personal Memory Brain) - local-first persistent memory for AI coding agents (MCP server, fully offline, no API keys)
TensorFlow IO
Lightweight version of Milvus for local development and testing
A library that prepares raw documents for downstream ML tasks.
tensorflow/datasets is a library of datasets ready to use with TensorFlow.
TensorFlow Hub is a library to foster the publication, discovery, and consumption of reusable parts of machine learning models.
A unified library for object tracking featuring clean room re-implementations of leading multi-object tracking algorithms
A gradient processing and optimization library in JAX.
Distrax: Probability distributions in JAX.
Self-hosted AI agents — streaming chat, tool use, persistent memory, multi-agent teams