Search results

[typing] Use ParamSpec in JIT annotation

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

Status: Open (in progress).

jax-ml/jaxnumber 14688

#14688 In jax-ml/jax;

· NeilGirdhar opened on Feb 26, 2023

improve tree mismatch error in tree_map

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

Status: Open (in progress).

jax-ml/jaxnumber 23771

#23771 In jax-ml/jax;

· mattjj opened on Sep 19, 2024

improve jnp.mean / jnp.sum / ... error message for out-of-bounds axis index

better_errorsImprove the error reportingImprove the error reporting

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

Status: Open (in progress).

jax-ml/jaxnumber 25155

#25155 In jax-ml/jax;

· mattjj opened on Nov 28, 2024

document class methods in autodiff autosummary

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

Status: Open (in progress).

jax-ml/jaxnumber 16639

#16639 In jax-ml/jax;

· froystig opened on Jul 6, 2023

Added xla_call and custom_jvp_call discharge rules

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

Status: Open (in progress).

jax-ml/jaxnumber 13837

#13837 In jax-ml/jax;

· patrick-kidger opened on Dec 31, 2022

Delay formatting of the name stack in translation rules

cla: yes

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

Status: Open (in progress).

jax-ml/jaxnumber 6889

#6889 In jax-ml/jax;

· apaszke opened on Jun 3, 2021

Clarify type hints for wrapped functions

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

Status: Open (in progress).

jax-ml/jaxnumber 25994

#25994 In jax-ml/jax;

· ncoghlan opened on Jan 21, 2025

Add type information for pallas bitcast/roll

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

Status: Open (in progress).

jax-ml/jaxnumber 20576

#20576 In jax-ml/jax;

· kkiningh opened on Apr 3, 2024

change grad(jnp.abs)(0.+0.j) convention to be 1.+0j

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

Status: Open (in progress).

jax-ml/jaxnumber 10800

#10800 In jax-ml/jax;

· mattjj opened on May 23, 2022

lobpcg: avoid changing X64 setting

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

Status: Open (in progress).

jax-ml/jaxnumber 21165

#21165 In jax-ml/jax;

· jakevdp opened on May 10, 2024

Add generic O(k^3) jet rule via truncated polynomial composition

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

Status: Open (in progress).

jax-ml/jaxnumber 13737

#13737 In jax-ml/jax;

· dsheldon opened on Dec 20, 2022

Fix debug_nans regressions.

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

Status: Open (in progress).

jax-ml/jaxnumber 25519

#25519 In jax-ml/jax;

· emilyfertig opened on Dec 16, 2024