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,436 Open 4,287 Closed
1,436 Open 4,287 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

time.sleep affects the execution time of JAX bug Something isn't working
#24941 opened Nov 18, 2024 by horse6
1
FloatingPointError in jax.scipy.stats bug Something isn't working
#24939 opened Nov 17, 2024 by IlayMenahem
@dfm
1
[GPU] FlashAttention performance lags behind PyTorch bug Something isn't working
#24934 opened Nov 17, 2024 by neel04
2
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
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
⚠️ Nightly upstream-dev CI failed ⚠️ CI
#24875 opened Nov 13, 2024 by github-actions bot
@jakevdp
1
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
Flaky test tests/fft_test.py::FftTest::testFftfreq5 bug Something isn't working
#24798 opened Nov 8, 2024 by apivovarov
@dougalm
2
Sparse reshape throws error when n_dense>0 and some target dimension has size 1 bug Something isn't working
#24795 opened Nov 8, 2024 by cherrywoods
1
Nightly builds on Mac ARM fail complex function numerical tests bug Something isn't working
#24787 opened Nov 8, 2024 by hawkinsp
2
Error with at[index].set(value) when using jit and shard_map in multi-host setting. bug Something isn't working
#24768 opened Nov 7, 2024 by lollcat
2
shard_map doesn't work with jnp.insert bug Something isn't working
#24762 opened Nov 7, 2024 by mrlazy1708
@mattjj
pure_callback using a threadpool to accelerate vmap enhancement New feature or request
#24756 opened Nov 7, 2024 by Joshuaalbert
3
Facing Scaling issue on cpu (arm and x86). bug Something isn't working
#24753 opened Nov 7, 2024 by choudhary-devang
ProTip! Mix and match filters to narrow down what you’re looking for.