Search results

[Pallas] Update emit_pipeline example.

kokoro:force-run

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

Status: Open (in progress).

jax-ml/jaxnumber 33386

#33386 In jax-ml/jax;

· justinjfu opened on Nov 18, 2025

[no-thunks] Avoid using linear util in <code>_initial_style_open_jaxpr</code>

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

Status: Open (in progress).

jax-ml/jaxnumber 33382

#33382 In jax-ml/jax;

· dougalm opened on Nov 17, 2025

fix ref effect indexing bugs

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

Status: Draft (not ready).

jax-ml/jaxnumber 33351

#33351 In jax-ml/jax;

· mattjj opened on Nov 16, 2025

Aot

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

Status: Open (in progress).

jax-ml/jaxnumber 33225

#33225 In jax-ml/jax;

· danielsuo opened on Nov 10, 2025

add <code>kind</code> to jax.new_ref (not just core.new_ref)

kokoro:force-run

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

Status: Open (in progress).

jax-ml/jaxnumber 33127

#33127 In jax-ml/jax;

· mattjj opened on Nov 5, 2025

Make tree_util.register_dataclass a decorator factory

kokoro:force-run

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

Status: Open (in progress).

jax-ml/jaxnumber 33070

#33070 In jax-ml/jax;

· jakevdp opened on Nov 3, 2025

Add Pallas core_map guide

kokoro:force-run

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

Status: Open (in progress).

jax-ml/jaxnumber 33048

#33048 In jax-ml/jax;

· IvyZX opened on Oct 31, 2025

wip remat3

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

Status: Open (in progress).

jax-ml/jaxnumber 33018

#33018 In jax-ml/jax;

· mattjj opened on Oct 30, 2025

Make <code>k</code> param of <code>jnp.eye</code> non-static.

kokoro:force-run

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

Status: Open (in progress).

jax-ml/jaxnumber 33012

#33012 In jax-ml/jax;

· infalmo opened on Oct 30, 2025

Fix use of deprecated jax_platform_name

kokoro:force-run

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

Status: Open (in progress).

jax-ml/jaxnumber 32840

#32840 In jax-ml/jax;

· Prince213 opened on Oct 23, 2025

Support explicit integer padding in lax.conv_transpose

kokoro:force-run

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

Status: Open (in progress).

jax-ml/jaxnumber 32268

#32268 In jax-ml/jax;

· samanklesaria opened on Oct 1, 2025

[hijax] vmap is not a primitive, it is sophisticated, hi-class

kokoro:force-run

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

Status: Open (in progress).

jax-ml/jaxnumber 32152

#32152 In jax-ml/jax;

· mattjj opened on Sep 26, 2025