AI/ML Frameworks

PyTorch, TensorFlow, JAX — run any framework on bare-metal NVIDIA GPUs.

PyTorch

Native CUDA kernels, DDP, FSDP, and Torch compile.

TensorFlow

Full TF GPU support with XLA and mixed precision.

JAX

Single-host and multi-host pod training with pmap.