[JAX] Expose sliding window attn to TE-JAX API #5601

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
JAX Python	31s
JAX C++	17s
PaddlePaddle C++	18s
PaddlePaddle Python	52s
PyTorch Python	1m 58s
PyTorch C++	22s
	4m 18s