Skip to content

jax.numpy reductions: avoid upcast of f16 when dtype is specified by user#26403

Draft
jakevdp wants to merge 1 commit intojax-ml:mainfrom jakevdp:bf16-mean
+5-4

Commits