Skip to content

Issues: jax-ml/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,489 Open 4,463 Closed
1,489 Open 4,463 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

[sharding_in_types] Unable to device_put an unsharded array inside use_mesh bug Something isn't working
#26552 opened Feb 15, 2025 by ayaka14732
1
@mattjj @dougalm @yashk2810
1
Request: Update jax-rocm60-plugin from 0.4.35 to 0.5 on PyPI to support jax[rocm] bug Something isn't working
#26510 opened Feb 13, 2025 by raj-magesh
@Ruturaj4
JAX internal functions must take a non-empty core.DebugInfo. better_errors Improve the error reporting
#26480 opened Feb 12, 2025 by gnecula
@gnecula
Make type checking strict under sharding-in-types bug Something isn't working
#26474 opened Feb 12, 2025 by yashk2810
@yashk2810
linalg.lstsq incorrectly returns nans bug Something isn't working
#26468 opened Feb 11, 2025 by carlosgmartin
Flash/Splash attention requires head_dim to be a multiple of 128, which is not compatible with deepseek's 192 head_dim enhancement New feature or request pallas Issues pertaining to Pallas (GPU or TPU)
#26433 opened Feb 10, 2025 by rodrigo-f-nogueira
@sharadmv @apaszke @Rifur13
Feature Request: Direct ONNX Exporter for JAX enhancement New feature or request
#26430 opened Feb 10, 2025 by johnnynunez
1
Unexpected host-to-device transfer when slicing question Questions for the JAX team
#26425 opened Feb 8, 2025 by alaurenzi
@jakevdp
3
Fix str + repr of multi-slice Devices enhancement New feature or request good first issue Good for newcomers
#26418 opened Feb 8, 2025 by skye
Windows builds enhancement New feature or request
#26417 opened Feb 7, 2025 by apasarkar
3
Break up lax_numpy.py cleanup Code cleanups and improvements
#26400 opened Feb 7, 2025 by jakevdp
@jakevdp
TSAN race in //tests:dynamic_api_test_cpu bug Something isn't working free threading Issues found in free threading builds
#26396 opened Feb 7, 2025 by hawkinsp
@vfdev-5
1
Parallelizing linalg.eig using sharding fails bug Something isn't working
#26379 opened Feb 7, 2025 by mfschubert
@dfm
4
jax.debug.print and jax.debug.breakpoint docs should mention perturbing the computation documentation
#26370 opened Feb 6, 2025 by mattjj
@alhridoy
1
Writing function input to global mutable array fails inside jax.grad bug Something isn't working
#26361 opened Feb 6, 2025 by ayaka14732
@mattjj @dougalm
Global mutable arrays do not work with AOT bug Something isn't working
#26354 opened Feb 6, 2025 by ayaka14732
@mattjj
jit(mutable_array) should raise an error bug Something isn't working
#26349 opened Feb 6, 2025 by ayaka14732
@mattjj
1
NaN gradient when using masked embeddings -> norm -> max bug Something isn't working
#26328 opened Feb 5, 2025 by VRichardJP
TSAN race in //tests:lax_numpy_einsum_test_cpu bug Something isn't working free threading Issues found in free threading builds
#26305 opened Feb 4, 2025 by hawkinsp
@vfdev-5
Rewriting the kernel in Triton hinders the overall optimization benefits of persistent caching bug Something isn't working
#26304 opened Feb 4, 2025 by MelodicDrumstep
@superbobry
jax.distributed with slurm/mpi not working on rocm AMD GPU Issues pertaining to AMD GPUs (ROCM)
#26298 opened Feb 4, 2025 by ksebaz
@JehandadKhan
2
Support for f-strings in debug.print enhancement New feature or request
#26296 opened Feb 4, 2025 by kcdodd
@sharadmv
4
jax profiler - trace events with absolute timestamps enhancement New feature or request
#26288 opened Feb 3, 2025 by james-aws
TSAN race in //tests:export_back_compat_test_cpu under free-threading bug Something isn't working free threading Issues found in free threading builds
#26272 opened Feb 3, 2025 by hawkinsp
@vfdev-5
2
Build fails due to missing visibility declaration for hipfft target AMD GPU Issues pertaining to AMD GPUs (ROCM) bug Something isn't working build
#26269 opened Feb 3, 2025 by TimoStolz
@JehandadKhan
3
ProTip! What’s not been updated in a month: updated:<2025-01-15.