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,304 Open 4,137 Closed
1,304 Open 4,137 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

JITing loss function causes erroneous results bug Something isn't working
#23226 opened Aug 24, 2024 by muchanem
Guard against unintentional transfers to CPU enhancement New feature or request
#23215 opened Aug 23, 2024 by kxs-dhenshall
2
Rotation.concatenate does not work for two single rotations bug Something isn't working
#23202 opened Aug 22, 2024 by thomas-rkk
@jakevdp
1
[Pallas GPU] jnp.abs() gives the wrong result bug Something isn't working
#23191 opened Aug 22, 2024 by ayaka14732
1
@ayaka14732
1
Pallas Unexpected Numeric Results With tf32 on A100 GPU bug Something isn't working
#23182 opened Aug 21, 2024 by axelfeldmann
@sharadmv
1
Pallas kernel crash at llo::CouldLtS32 when interpret=False on TPU bug Something isn't working
#23181 opened Aug 21, 2024 by nathom
@sharadmv @justinjfu
3
Unsupported conversion from f64 to f16 in pallas despite not using fp16 bug Something isn't working
#23179 opened Aug 21, 2024 by lengstrom
@superbobry
1
erf_inv not supported in Pallas GPU bug Something isn't working
#23173 opened Aug 21, 2024 by ayaka14732
1
@ayaka14732
Improve support for using JAX's custom LAPACK calls externally enhancement New feature or request
#23172 opened Aug 21, 2024 by joeycarter
2
jax library error jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed bug Something isn't working
#23145 opened Aug 20, 2024 by MikaBell
3
JAX segment_sum is two times slower for FP16 inputs than FP32 inputs bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs XLA
#23136 opened Aug 20, 2024 by CloudyDory
10
CuDNN Initialization Error with Pre-existing CUDA Environment Fails with bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#23097 opened Aug 16, 2024 by parambole
@nouiz
5
Performance Discrepancy in jax.value_and_grad for Different argnums bug Something isn't working
#23085 opened Aug 16, 2024 by demon2036
6
Explicit managing the output buffer of jax.jit function enhancement New feature or request
#23084 opened Aug 15, 2024 by shawnwang18
@jakevdp
1
Add support nogil mode in JAX for Python 3.13 enhancement New feature or request
#23073 opened Aug 15, 2024 by vfdev-5
1 task
@vfdev-5
custom_jvp leaks tracers if they're marked as nondiff_argnum better_errors Improve the error reporting
#23065 opened Aug 14, 2024 by jakevdp
@froystig @jakevdp
2
Port all custom calls to FFI enhancement New feature or request
#23056 opened Aug 14, 2024 by dfm
5 of 20 tasks
@dfm
Scatter on Sharded Matrices has bugs bug Something isn't working
#23052 opened Aug 14, 2024 by ymahlau
1
Document Named Collectives documentation
#23024 opened Aug 12, 2024 by lockwo
@jakevdp
1
Errors from interaction between shard_map with auto and lax.map bug Something isn't working
#23019 opened Aug 12, 2024 by jeffgortmaker
@pschuh
Finite precision error calculations always 0 under JIT with bfloat16 bug Something isn't working
#23007 opened Aug 12, 2024 by colehaus
@jakevdp
3
CUDA XlaRuntimeError with MPI on jax==0.4.31 bug Something isn't working
#22995 opened Aug 12, 2024 by MasterSkepticista
5
Accessing incoming gradient and activations in custom_vjp enhancement New feature or request
#22993 opened Aug 12, 2024 by yidingjiang
@dfm
4
Complex gradient question Questions for the JAX team
#22948 opened Aug 8, 2024 by inversecrime
@mattjj
3
Extending JAX with CUDA: cstdint: error: the global scope has no "..." bug Something isn't working
#22946 opened Aug 8, 2024 by bsaoptima
@dfm
3
ProTip! Type g i on any issue or pull request to go back to the issue listing page.