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,492 Open 4,439 Closed
1,492 Open 4,439 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

TSAN race in //tests:shard_map_test_cpu bug Something isn't working free threading Issues found in free threading builds
#26359 opened Feb 6, 2025 by hawkinsp
@hawkinsp
1
Global mutable arrays do not work with AOT bug Something isn't working
#26354 opened Feb 6, 2025 by ayaka14732
@mattjj
ref_get mutates the sharding of the underlying buffer of a mutable array bug Something isn't working
#26350 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
JAX numpy linalg gives different results from standard numpy linalg question Questions for the JAX team
#26347 opened Feb 6, 2025 by Shaoqigit
@jakevdp
1
Creating a mutable array does not preserve its sharding bug Something isn't working
#26338 opened Feb 5, 2025 by ayaka14732
@mattjj
2
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
2
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
Grouped convolutions are much slower than standard convolutions when increasing kernel dilation bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs performance make things lean and fast
#26266 opened Feb 2, 2025 by karunrao97
@mjsML
1
Support Bounded Shape Input enhancement New feature or request
#26265 opened Feb 2, 2025 by yaochengji
@mattjj
Gradient of jnp.linalg.norm does not respect absolute homogeneity. bug Something isn't working
#26248 opened Jan 31, 2025 by tillahoffmann
@mattjj
1
jax.ops.segment_sum is 100x slower when segment_ids are sorted, but only for float16 bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#26227 opened Jan 31, 2025 by rafaelha
@pifon2a
2
Document jax.lax APIs documentation
#26219 opened Jan 30, 2025 by jakevdp
@jakevdp
Control the lifetime of host values of the jax.Array. bug Something isn't working
#26216 opened Jan 30, 2025 by yliu120
@dfm
10
Document more practical use-cases for vmap and cases that cannot be covered by additional batch axes enhancement New feature or request
#26212 opened Jan 30, 2025 by adonath
@jakevdp
1
Issues building from source on an M3 chip bug Something isn't working
#26207 opened Jan 30, 2025 by botev-openai
6
Jaxpr string representation of dataclasses enhancement New feature or request
#26202 opened Jan 30, 2025 by botev
@jakevdp
1
Add documentation about behavior of linear solvers in case the system matrix is singular enhancement New feature or request
#26180 opened Jan 29, 2025 by NKlug
2
Operation very slow to compile since jax 0.4.36 bug Something isn't working
#26162 opened Jan 28, 2025 by btaba
8
ProTip! What’s not been updated in a month: updated:<2025-01-06.