3 ms·Reducing HBM Bottlenecks in JAX-Based LLM Training with Host Offloading2 points by matt_d 3mo ago