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

Rank-one updates to eigenvalue decompositions enhancement New feature or request
#25057 opened Nov 22, 2024 by mishavanbeek
1
jax.numpy.linalg.multi_dot is O(2^N) in the number of matrices being multiplied bug Something isn't working
#25051 opened Nov 22, 2024 by rohan-hitchcock
5
[Pallas] Uable to convert negative values from float16/float32 to int8/int32 in pallas bug Something isn't working
#25047 opened Nov 21, 2024 by shangz-ai
Running the executable compiled directly from jax.jit is more than three times slower than jax.jit itself. bug Something isn't working
#25023 opened Nov 21, 2024 by caixiiaoyang
1
How do you remat GSPMD inserted all-gathers? question Questions for the JAX team
#25010 opened Nov 20, 2024 by ptoulme-aws
@mattjj
1
CUDA_ERROR_ILLEGAL_ADDRESS bug Something isn't working
#25002 opened Nov 20, 2024 by PhilipVinc
Add a way to get the tree size of a jaxpr enhancement New feature or request
#24995 opened Nov 20, 2024 by carlosgmartin
7
An example for ffi input/output aliasing enhancement New feature or request
#24986 opened Nov 19, 2024 by saeedmaleki
1
@dfm
1
AOT compilation and serialization bug Something isn't working
#24982 opened Nov 19, 2024 by jkup14
13
debug_nans error always says the de-optimized function did not produce NaNs bug Something isn't working
#24955 opened Nov 18, 2024 by emilyfertig
@mattjj
5
[GPU] FlashAttention performance lags behind PyTorch bug Something isn't working
#24934 opened Nov 17, 2024 by neel04
4
Small Einsum is hanging bug Something isn't working
#24929 opened Nov 16, 2024 by ryan112358
@jakevdp
2
Rework comparisons in the docs of JAX vs. NumPy PRNG documentation
#24927 opened Nov 16, 2024 by emilyfertig
1
Allow einsum to support naive contraction strategy enhancement New feature or request
#24915 opened Nov 15, 2024 by ryan112358
1
Results do not match the reference. This is likely a bug/unexpected loss of precision bug Something isn't working
#24909 opened Nov 15, 2024 by yanboyang97
CUDA12 plugin segfaults when older version of JAX is installed bug Something isn't working
#24901 opened Nov 14, 2024 by dime10
Missing annotations bug Something isn't working
#24888 opened Nov 13, 2024 by PerilousApricot
@jakevdp
4
array.at.set is incredibly slow for complex128 dtype bug Something isn't working
#24872 opened Nov 13, 2024 by chrisrothUT
5
AOT errors when setting compile options bug Something isn't working
#24869 opened Nov 12, 2024 by man2machine
Unsupported type in metal PJRT plugin with rng_bit_generator Apple GPU (Metal) plugin bug Something isn't working
#24867 opened Nov 12, 2024 by dmarro89
@kulinseth @shuhand0
CUDA_ERROR_SYSTEM_NOT_READY: system not yet initialized bug Something isn't working
#24866 opened Nov 12, 2024 by carlosgmartin
1
Deepcopy of pjit functions failing (or not supported) bug Something isn't working
#24838 opened Nov 11, 2024 by jlperla
support q offset w.r.t k/v in flash_attention function enhancement New feature or request
#24830 opened Nov 11, 2024 by yamingx
2
potential jax 0.4.35 release issue? bug Something isn't working
#24826 opened Nov 10, 2024 by haohuanw
@mattjj
1
Division by self not always "1.0" on JAX GPU, but consistently gives "1.0" on JAX CPU. bug Something isn't working
#24807 opened Nov 8, 2024 by mattlevine22
1
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.