NVIDIA Optimizes JAX LLM Training with Host Offloading - TrendCloud