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

Clear current search query, filters, and sorts
1,415 Open 4,247 Closed
1,415 Open 4,247 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

Functions like jax.numpy.concatenate are not compatible with jax.sharding, resulting in improper functioning bug Something isn't working
#24576 opened Oct 29, 2024 by kYangLi
jax.scan as a function transformation enhancement New feature or request
#24573 opened Oct 29, 2024 by smorad
2 tasks done
MaxText Getting Started fails bug Something isn't working
#24549 opened Oct 28, 2024 by learning-to-play
1
segmentation fault from a simple test on GCP vm bug Something isn't working
#24548 opened Oct 27, 2024 by jiyuuchc
2
New function to allow easy paralelization over vmap() enhancement New feature or request
#24547 opened Oct 27, 2024 by MRiabov
1 task done
@jakevdp
4
⚠️ Nightly upstream-dev CI failed ⚠️ CI
#24544 opened Oct 26, 2024 by github-actions bot
@dfm
2
return in finally swallows exceptions bug Something isn't working
#24541 opened Oct 25, 2024 by iritkatriel
1
[Pallas][Flash Attention] Cost analysis in Flash Attention kernel causing compilation OOM bug Something isn't working
#24539 opened Oct 25, 2024 by lsy323
INVALID_ARGUMENT when copying array in pinned_host memory with device_put(..., may_alias=False) bug Something isn't working
#24521 opened Oct 25, 2024 by ojw28
hlo_helpers.dtype_to_ir_type is not complete comparing with jax.interpreters.mlir.dtype_to_ir_type enhancement New feature or request
#24509 opened Oct 24, 2024 by aws-qieqingy
@dfm
7
pscatter function enhancement New feature or request
#24506 opened Oct 24, 2024 by luyug
Get wrong structure when using jax2tf to convert nnx.module into tflite file bug Something isn't working
#24497 opened Oct 24, 2024 by noahzhy
1
Matmul operandlhs must have a shape divisible by (16, 1), but got: (8, 2048) bug Something isn't working
#24486 opened Oct 23, 2024 by vanbasten23
Attribute Error bug Something isn't working
#24478 opened Oct 23, 2024 by Opiyo-Inno
@dougalm
17
[Pallas TPU] Unable to create a boolean array inside kernel due to Mosaic relayout error bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#24464 opened Oct 22, 2024 by ayaka14732
@ayaka14732
1
Full coverage for __jax_array__ protocol bug Something isn't working
#24460 opened Oct 22, 2024 by maxencefaldor
@mattjj
1
[ROCM] Sharding leads to HIP errors bug Something isn't working
#24458 opened Oct 22, 2024 by PhilipVinc
Inconsistency in using == operator on a jax array and a list bug Something isn't working
#24454 opened Oct 22, 2024 by gokul-uf
@jakevdp
1
pmap of pjit enhancement New feature or request
#24435 opened Oct 21, 2024 by luyug
4
shard_map - NotImplementedError: No replication rule for cond bug Something isn't working
#24418 opened Oct 21, 2024 by giovannicemin
2
Not implemented: Non-trivial layouts unsupported bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#24415 opened Oct 20, 2024 by vanbasten23
4
jax.lax.approx_min_k() on GPU appears to be a brute force search bug Something isn't working
#24414 opened Oct 20, 2024 by notnot
JAX code is extremely slow on GPUs bug Something isn't working performance make things lean and fast
#24411 opened Oct 20, 2024 by AakashKumarNain
@mattjj
12
docs: pprof command is out of date bug Something isn't working
#24407 opened Oct 19, 2024 by garymm
1
INTERNAL: Mosaic failed to compile TPU kernel: unsupported shape cast bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#24404 opened Oct 19, 2024 by vanbasten23
4
ProTip! Mix and match filters to narrow down what you’re looking for.