Lint

[JAX] Flax params initialization with weight_dtype #6924

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
JAX C++	23s
PyTorch C++	21s
JAX Python	19s
PyTorch Python	1m 46s
	2m 49s