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
Build failures at HEAD on Apple ARM64
bug
Something isn't working
#10132
opened Apr 4, 2022 by
nicholasjng
1 task done
JIT changes the output of a function
bug
Something isn't working
#10129
opened Apr 2, 2022 by
fbartolic
NotImplementedError: outfeed rewrite custom_linear_solve
enhancement
New feature or request
#10112
opened Mar 31, 2022 by
denizokt
host_callback.call(..., call_with_device=True) specifies the wrong device
bug
#10082
opened Mar 30, 2022 by
patrick-kidger
Adding the corresponding indexes in New feature or request
P3 (no schedule)
We have no plan to work on this and, if it is unassigned, we would be happy to review a PR
jax.ops.segment_max?
enhancement
#10079
opened Mar 30, 2022 by
felixchalumeau
Add environment variable for setting absl-py logging level
enhancement
New feature or request
#10070
opened Mar 29, 2022 by
skye
jnp.linalg.svd etc. do not respect __jax_array__
contributions welcome
#10065
opened Mar 29, 2022 by
patrick-kidger
Proposal New feature or request
negative_as_oob for ndarray.at[...]
enhancement
#10063
opened Mar 29, 2022 by
JeppeKlitgaard
jax svd is exceptionally slow on TPUs
enhancement
New feature or request
P2 (eventual)
This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#10052
opened Mar 28, 2022 by
rgreenblatt
Document best practices/requirements for writing JAX libraries
documentation
enhancement
New feature or request
P2 (eventual)
This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#10047
opened Mar 28, 2022 by
FedericoV
Proposal - jax.jit static_argnums and static_argnames raises warning if parameter doesn't exist
enhancement
New feature or request
#10046
opened Mar 28, 2022 by
pseudo-rnd-thoughts
jax2tf fails with static_arguments
bug
Something isn't working
#10044
opened Mar 28, 2022 by
sbodenstein
Add more default arguments
enhancement
New feature or request
#10035
opened Mar 25, 2022 by
carlosgmartin
6 tasks
lax.while_loop within jax.disable_jit() should not silently fall back to the primitive version.
enhancement
New feature or request
#10023
opened Mar 24, 2022 by
YouJiacheng
jax.lax.custom_linear_solve / jax.lax.custom_root doesn't respect jax.disable_jit
enhancement
New feature or request
#10022
opened Mar 24, 2022 by
YouJiacheng
Cannot Share GPUs between processes when using JAX with SLURM
bug
Something isn't working
#10021
opened Mar 24, 2022 by
chrisgrimm
flake8 ignore statements are incorrect
P2 (eventual)
This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#10014
opened Mar 23, 2022 by
jakevdp
vjp TypeError on weakly typed cotangents
bug
Something isn't working
P2 (eventual)
This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#9994
opened Mar 22, 2022 by
eelregit
jax2tf shape polymorphism fails with slice
bug
Something isn't working
#9974
opened Mar 21, 2022 by
sbodenstein
jax.lax.conv_general_dilated_patches returns nan for patches with inf
bug
#9972
opened Mar 21, 2022 by
yovadia
bug with jit, custom_vjp and reduce_window
bug
Something isn't working
#9961
opened Mar 18, 2022 by
mariogeiger
Add control flow operators for iterated functions
enhancement
New feature or request
#9960
opened Mar 18, 2022 by
carlosgmartin
Modified Bessel function of the second kind in jax.scipy.special
enhancement
New feature or request
#9956
opened Mar 18, 2022 by
egorssed
Previous Next
ProTip!
What’s not been updated in a month: updated:<2022-03-04.

