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,430 Open 4,285 Closed
1,430 Open 4,285 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

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
3
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
Bug in the latest libtpu nightly release bug Something isn't working
#24829 opened Nov 11, 2024 by knyazer
@emilyfertig
1
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
1
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
2
Facing Scaling issue on cpu (arm and x86). bug Something isn't working
#24753 opened Nov 7, 2024 by choudhary-devang
cudaErrorSymbolNotFound : named symbol not found bug Something isn't working
#24749 opened Nov 6, 2024 by amanjitsk
3
scipy.sparse.csgraph.connected_components implementation enhancement New feature or request
#24737 opened Nov 6, 2024 by vboussange
@jakevdp
2
TPU hangs if using random inside a pure_callback call bug Something isn't working
#24720 opened Nov 5, 2024 by tengomucho
5
Allow specifying exec_time_optimization_effort and memory_fitting_effort via CLI. enhancement New feature or request
#24715 opened Nov 5, 2024 by golechwierowicz
@mattjj
Disable 64 bit codegen to HLO for Neuron backend enhancement New feature or request
#24682 opened Nov 4, 2024 by ptoulme-aws
7
DLPack Support for np.uint16 seems broken bug Something isn't working
#24680 opened Nov 3, 2024 by realquantumcookie
2
[Pallas] Unable to Modify Input Ref in Pallas Kernel bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#24656 opened Nov 1, 2024 by shangz-ai
3
ProTip! What’s not been updated in a month: updated:<2024-10-13.