New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Author
Label
Projects
Milestones
Assignee
Sort
"Disabling cuDNN frontend for the following convolution" and warning when upgrade jax
bug
#9460
opened Feb 6, 2022 by
wztdream
jax.experimental.host_callback.call has different behaviour on CPU vs GPU, for exceptions
bug
#9457
opened Feb 5, 2022 by
patrick-kidger
"DNN library is not found." error when tensorflow is loaded before JAX
bug
#9454
opened Feb 5, 2022 by
samuela
3 tasks done
TypeError: No constant handler for type: <class 'jax.core.Token'>
bug
#9438
opened Feb 3, 2022 by
Thenerdstation
grad of grad fails when ravelling is involved
question
#9412
opened Feb 2, 2022 by
gianlucadetommaso
Unexpected result when using jax.jit and returning array and its transpose.
bug
xla_issue
#9399
opened Feb 1, 2022 by
eugenhu
jax.custom_jvp(nondiff_argnums=...) crashes inside of lax.cond.
bug
#9374
opened Jan 29, 2022 by
patrick-kidger
Significant performance loss caused by matrix indexing in jitted function with nested jax operations
bug
needs info
performance
#9366
opened Jan 28, 2022 by
yannra
CTC Loss for Speech Recognition models
enhancement
#9350
opened Jan 27, 2022 by
patrickvonplaten
2 tasks done
Documentation for
jax.checkpoint(concrete=..., prevent_cse=...)
documentation
#9348
opened Jan 27, 2022 by
patrick-kidger
A case in which
jax.jit changes results, prevents backprop, and inhibits optimisations
bug
#9298
opened Jan 24, 2022 by
patrick-kidger
Efficient argmin/argmax for bool type arrays
enhancement
jax.numpy
P3 (no schedule)
performance
xla_issue
#9282
opened Jan 21, 2022 by
Joshuaalbert
jax.numpy.nanpercentile with axis as tuple
contributions welcome
enhancement
jax.numpy
#9277
opened Jan 21, 2022 by
clee1994
Jax GMRES on GPU largely slower than its scipy counterpart
performance
xla_issue
#9259
opened Jan 20, 2022 by
Azercoco
providing axis_size to pmap should remove the need to infer it from args
bug
#9252
opened Jan 20, 2022 by
mattjj
compilation cache: get/put with
Lowered and Compiled instances
enhancement
P2 (eventual)
#9250
opened Jan 19, 2022 by
froystig
Previous Next
ProTip!
no:milestone will show everything without a milestone.

