Closed gobbleturk closed 2 months ago
Test failing with new jax release - will disable for now while we investigate and fix
Although this AOT test fails with OOM that the tensor is in HBM instead of host, runs on real hardware work fine, activations are offloaded to host.
Internal tracking: b/366200617
Test failing with new jax release - will disable for now while we investigate and fix
Although this AOT test fails with OOM that the tensor is in HBM instead of host, runs on real hardware work fine, activations are offloaded to host.
Internal tracking: b/366200617