Search results

add more detail about aux_data in pytree docs

cla: yes

Status: Open (in progress).

jax-ml/jaxnumber 8739

#8739 In jax-ml/jax;

· mattjj opened on Nov 30, 2021

Improve error message when passing a tuple of non-concrete values as …

better_errorsImprove the error reportingImprove the error reporting

cla: yes

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

Status: Open (in progress).

jax-ml/jaxnumber 7952

#7952 In jax-ml/jax;

· hawkinsp opened on Sep 17, 2021

[RFC] Add solver diagnostics for cg and bicgstab

cla: yes

Status: Draft (not ready).

jax-ml/jaxnumber 7825

#7825 In jax-ml/jax;

· PhilipVinc opened on Sep 6, 2021

In some cases axis_name is a tuple instead of string, handling it in …

cla: yes

Status: Open (in progress).

jax-ml/jaxnumber 7393

#7393 In jax-ml/jax;

· ahmedjawedaj opened on Jul 27, 2021

Fix unused variable and warnings in docs Neural Networks Tutorial

cla: yes

Status: Open (in progress).

jax-ml/jaxnumber 7333

#7333 In jax-ml/jax;

· FelipeMartins96 opened on Jul 20, 2021

Remove TraceContext & StepTraceContext

cla: yes

Status: Open (in progress).

jax-ml/jaxnumber 7050

#7050 In jax-ml/jax;

· kaushikb11 opened on Jun 22, 2021

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

Remat extensions for scan-over-layers

cla: yes

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

Status: Draft (not ready).

jax-ml/jaxnumber 6639

#6639 In jax-ml/jax;

· jekbradbury opened on May 4, 2021

FFT Convolution

cla: yes

Status: Open (in progress).

jax-ml/jaxnumber 6343

#6343 In jax-ml/jax;

· peterroelants opened on Apr 5, 2021

add more options for jax_default_matmul_precision

cla: yes

Status: Open (in progress).

jax-ml/jaxnumber 6217

#6217 In jax-ml/jax;

· mattjj opened on Mar 24, 2021

Adding a new transposed convolution function to lax

cla: yes

Status: Open (in progress).

jax-ml/jaxnumber 5772

#5772 In jax-ml/jax;

· yang-song opened on Feb 18, 2021

add pargmin_p / pargmax_p primitives

cla: yes

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

Status: Open (in progress).

jax-ml/jaxnumber 5694

#5694 In jax-ml/jax;

· mattjj opened on Feb 10, 2021