jacrev, jacfwd and grad do not propagate sharding #25844 opened Jan 10, 2025 Better jax.debug.print #25842 opened Jan 10, 2025 Got an error when offloading SplashAttention pallas call #25841 opened Jan 10, 2025 RnnTest.test_struct_encoding_determinism failing #25825 opened Jan...