Distributed Model Training Tools

JAX

AI Framework

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

Apache Spark MLlib

Development

The industry-standard distributed machine learning library for ultra-scale big data processing.

Together AI Platform

AI Infrastructure

AI Native Cloud for training, fine-tuning, and inference of open-source and specialized models.