Skip to content

Small fix to prevent OOM in Imagenet VIT - #883

Merged
priyakasimbeg merged 1 commit into
devfrom
jit_switch
Aug 21, 2025
Merged

Small fix to prevent OOM in Imagenet VIT#883
priyakasimbeg merged 1 commit into
devfrom
jit_switch

remove jax.device_put from imagenet test pipeline because it results …

cfbaf7a
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar