Search results

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

Improve error message for grad-of-eig

better_errorsImprove the error reportingImprove the error reporting

Status: Open (in progress).

jax-ml/jaxnumber 24350

#24350 In jax-ml/jax;

· jakevdp opened on Oct 16, 2024

Better errors for traced nondiff_argnums

better_errorsImprove the error reportingImprove the error reporting

Status: Open (in progress).

jax-ml/jaxnumber 23066

#23066 In jax-ml/jax;

· jakevdp opened on Aug 14, 2024

custom_jvp leaks tracers if they're marked as nondiff_argnum

better_errorsImprove the error reportingImprove the error reporting

Status: Open.

#23065 In jax-ml/jax;

· jakevdp opened on Aug 14, 2024

Improve error messages for invalid indices

better_errorsImprove the error reportingImprove the error reporting

Status: Open.

#22798 In jax-ml/jax;

· jakevdp opened on Jul 31, 2024

CTRL+C Broken When Running distributed.initialize() on one TPU Host by Accident

better_errorsImprove the error reportingImprove the error reporting

Status: Open.

#22436 In jax-ml/jax;

· s-smits opened on Jul 14, 2024

improve error message and docs for custom_jvp / custom_vjp nondiff_argnums escaped tracer error

better_errorsImprove the error reportingImprove the error reporting

documentation

Status: Open.

#20889 In jax-ml/jax;

· mattjj opened on Apr 23, 2024

Pallas Kernel using Smem/SReg failed to lower

better_errorsImprove the error reportingImprove the error reporting

pallasIssues pertaining to Pallas (GPU or TPU)Issues pertaining to Pallas (GPU or TPU)

Status: Open.

#20046 In jax-ml/jax;

· tankbattle opened on Mar 1, 2024

Transpose rule for 'custom_vjp_call_jaxpr' not implemented for grad(custom_jvp(custom_vjp))

better_errorsImprove the error reportingImprove the error reporting

Status: Open.

#19087 In jax-ml/jax;

· patrick-kidger opened on Dec 21, 2023

silent truncation of numpy int64 arrays to int32 range

P2 (eventual)This ought to be addressed, but has no schedule at the moment. (Assignee optional)This ought to be addressed, but has no schedule at the moment. (Assignee optional)

better_errorsImprove the error reportingImprove the error reporting

Status: Open.

#18385 In jax-ml/jax;

· mattjj opened on Nov 3, 2023

jax.tree_utils.tree_map raises errors for dictionary with mixed types of keys

better_errorsImprove the error reportingImprove the error reporting

documentation

Status: Open.

#15358 In jax-ml/jax;

· zi-w opened on Apr 1, 2023

better errors during transposition

better_errorsImprove the error reportingImprove the error reporting

enhancementNew feature or requestNew feature or request

Status: Open.

#13455 In jax-ml/jax;

· froystig opened on Nov 30, 2022