dockers/base-xla - Lightning-AI/pytorch-lightning

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

0 0 0 Python

	Dockerfile	4.3 KB
	tpu_workflow_lite.jsonnet	2.0 KB
	tpu_workflow_pytorch.jsonnet	2.0 KB