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

1,002 Open 2,829 Closed
1,002 Open 2,829 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
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

jax.device_put not working under multiprocessing bug Something isn't working
#14184 opened Jan 27, 2023 by vwxyzjn
Add complex32 dtype support enhancement New feature or request
#14180 opened Jan 27, 2023 by magnusdk
2 tasks done
1
gradient of segment_sum gives nan bug Something isn't working
#14179 opened Jan 27, 2023 by thisiscam
autodidax fails with jaxlib 0.4.2 bug Something isn't working documentation
#14171 opened Jan 27, 2023 by jakevdp
1
@jakevdp
Build Fail with "fatal error LNK1169: one or more multiply defined symbols found" bug Something isn't working
#14165 opened Jan 26, 2023 by adam-hartshorne
3
Precision control has no effect bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs XLA
#14157 opened Jan 25, 2023 by yhtang
@pschuh
4
VJP error when custom_vjp and custom_vmap are used together. bug Something isn't working
#14150 opened Jan 25, 2023 by chr1sj0nes
@froystig
1
Inconsistency between jacfwd and jacrev for functions with multiple outputs bug Something isn't working
#14149 opened Jan 25, 2023 by VedPatwardhan
@mattjj
⚠️ Nightly upstream-dev CI failed ⚠️ CI
#14107 opened Jan 21, 2023 by github-actions bot
@jakevdp
1
Using lpmn_values for Legendre polynomials. contributions welcome The JAX team has not prioritized work on this. Community contributions are welcome. enhancement New feature or request
#14101 opened Jan 20, 2023 by BeeGass
jit compilation with large inlined arrays crashes JAX. bug Something isn't working needs info More information is required to diagnose & prioritize the issue.
#14080 opened Jan 19, 2023 by ethanluoyc
@pschuh
11
Increased memory requirement without nn.scan GPU Issues that apply to both CUDA and ROCM GPUs NVIDIA GPU Issues specific to NVIDIA GPUs performance make things lean and fast XLA
#14056 opened Jan 18, 2023 by abhinavgoel95
2
How do I reproduce test cases with the new custom PRNG? enhancement New feature or request
#14046 opened Jan 17, 2023 by NeilGirdhar
@froystig
3
Precision argument for matmul is not respected on Ampere GPUs like the RTX 3090 bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#14022 opened Jan 15, 2023 by jaschau
@jprabhas
jax.scipy.linalg.eigh_tridiagonal() doesn't implement calculation of eigenvectors enhancement New feature or request
#14019 opened Jan 15, 2023 by crowsonkb
5
XlaRuntimeError when jitting certain reductions of large (intermediary) array bug Something isn't working needs info More information is required to diagnose & prioritize the issue. NVIDIA GPU Issues specific to NVIDIA GPUs XLA
#14013 opened Jan 14, 2023 by deasmhumhna
4
Bazel Build Failed because it can't identify Python path bug Something isn't working contributions welcome The JAX team has not prioritized work on this. Community contributions are welcome. Windows Issues related to JAX on Microsoft Windows
#14012 opened Jan 14, 2023 by timelessnesses
2
test_nondiff_arg_jit_tracer test in api_test.py is leaking tracers with initial style transformation like pjit, scan bug Something isn't working
#14009 opened Jan 13, 2023 by yashk2810
@mattjj
1
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
Jax should expose options to disable CSE. enhancement New feature or request
#13999 opened Jan 13, 2023 by nshepperd
@pschuh
7
Document how to run unit tests on Cloud TPU enhancement New feature or request P2 (eventual) This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#13989 opened Jan 12, 2023 by skye
@skye
jax.scipy.linalg.solve_triangular gives false result with GPU bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#13979 opened Jan 12, 2023 by dkn16
@hawkinsp
2
Can't initialize cloud TPU on GKE TPU Nodes bug Something isn't working
#13969 opened Jan 11, 2023 by zurcnilva213
@skye
2
[jax2tf] Behavior of dimension polynomials as hashables enhancement New feature or request
#13964 opened Jan 11, 2023 by gnecula
@gnecula
enforcing eigendecomposition on CPU in TPU process bug Something isn't working
#13959 opened Jan 11, 2023 by yonghakim
@skye
1
ProTip! Adding no:label will show everything without a label.