Skip to content

Run the PyTorch model in bfloat16 to match the JAX compute dtype - #23

Merged
weihaokong merged 2 commits into
mainfrom
run-pytorch-in-bf16
Jul 2, 2026
Merged

Run the PyTorch model in bfloat16 to match the JAX compute dtype#23
weihaokong merged 2 commits into
mainfrom
run-pytorch-in-bf16

Note that the dtype option may be removed in a future release

7bbb904
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar