Search results

fused api prototype for fine-grained memory control

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Open (in progress).

jax-ml/jaxnumber 30931

#30931 In jax-ml/jax;

· mattjj opened on Aug 12, 2025

[debug-info] make debug info optional again, drop where it's incorrect

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Open (in progress).

jax-ml/jaxnumber 30868

#30868 In jax-ml/jax;

· mattjj opened on Aug 9, 2025

Add documentation for <code>jax.experimental.colocated_python</code>

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Open (in progress).

jax-ml/jaxnumber 30854

#30854 In jax-ml/jax;

· hyeontaek opened on Aug 8, 2025

core.Tracer shouldn't inherit from basearray.Array

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Open (in progress).

jax-ml/jaxnumber 30487

#30487 In jax-ml/jax;

· mattjj opened on Jul 24, 2025

Document all checkpoint policies in one place, on the JAX public API page.

documentation

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Open (in progress).

jax-ml/jaxnumber 30177

#30177 In jax-ml/jax;

· carlosgmartin opened on Jul 12, 2025

Add array-api copy semantics to DLPackManagedTensorToBuffer

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Open (in progress).

jax-ml/jaxnumber 29963

#29963 In jax-ml/jax;

· pearu opened on Jul 3, 2025

Add nd_loop and Enable block_n tiling for all_gather_lhs_matmul

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Open (in progress).

jax-ml/jaxnumber 29822

#29822 In jax-ml/jax;

· hanzlfs opened on Jun 27, 2025

Add support for unroll=0 to lax.scan.

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Open (in progress).

jax-ml/jaxnumber 29630

#29630 In jax-ml/jax;

· carlosgmartin opened on Jun 21, 2025

Add <code>split_p</code> to <code>jax.extend.core.primitives</code>

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Open (in progress).

jax-ml/jaxnumber 29452

#29452 In jax-ml/jax;

· j-towns opened on Jun 13, 2025

Update missed ml_dtypes version check to 0.5.0

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Open (in progress).

jax-ml/jaxnumber 29429

#29429 In jax-ml/jax;

· DanisNone opened on Jun 12, 2025

added solve_sylvester and accompanying tests

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Open (in progress).

jax-ml/jaxnumber 28810

#28810 In jax-ml/jax;

· Dekermanjian opened on May 17, 2025

Deprecate arr.T property for arrays with ndim != 2

kokoro:force-run

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Draft (not ready).

jax-ml/jaxnumber 28798

#28798 In jax-ml/jax;

· jakevdp opened on May 16, 2025