🧬

JAX

1 test cases

Google's JAX framework with PaxML for distributed training leveraging XLA compilation and automatic parallelism strategies on GPU clusters.

🧬

JAX PaxML

JAX-based distributed training with Google Pax framework

JAXPaxMLXLAAuto-parallelism