Skip to content

Issues: google/jax

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.

Sign up for GitHub

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

Clear current search query, filters, and sorts
8 Open 14 Closed
8 Open 14 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Issues list

cuDNN RNNs fail if any sequence lengths are 0 bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#17966 opened Oct 5, 2023 by hawkinsp
@nouiz
1
Slurm initialization only supports one device per host enhancement New feature or request NVIDIA GPU Issues specific to NVIDIA GPUs
#16788 opened Jul 19, 2023 by Findus23
@nouiz
8
jnp.save in a distributed mannor through GPUDirect enhancement New feature or request NVIDIA GPU Issues specific to NVIDIA GPUs
#15281 opened Mar 29, 2023 by sh0416
2 tasks done
@nouiz
4
jax[cuda]=0.4.1 version is slower on A100 GPU bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#14002 opened Jan 13, 2023 by patel-zeel
@nouiz
1
Faster FP32 power calculation on CUDA GPUs enhancement New feature or request NVIDIA GPU Issues specific to NVIDIA GPUs
#12509 opened Sep 26, 2022 by yhtang
@nouiz @yhtang
13
Bogus gradient value from value_and_grad for committed DeviceArray on a multi-GPU host bug Something isn't working needs info More information is required to diagnose & prioritize the issue. NVIDIA GPU Issues specific to NVIDIA GPUs
#11478 opened Jul 13, 2022 by yiiyama
@nouiz
11
Extremely slow GPU execution bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs P1 (soon) Assignee is working on this now, among other tasks. (Assignee required)
#7024 opened Jun 18, 2021 by NeilGirdhar
@nouiz
30
Gradients with odeint slow on GPU NVIDIA GPU Issues specific to NVIDIA GPUs P1 (soon) Assignee is working on this now, among other tasks. (Assignee required)
#5006 opened Nov 24, 2020 by spenrich
@nouiz
5
ProTip! Mix and match filters to narrow down what you’re looking for.