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.

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

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

Issues list

Allow einsum to support naive contraction strategy enhancement New feature or request
#24915 opened Nov 15, 2024 by ryan112358
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
array.at.set is incredibly slow for complex128 dtype bug Something isn't working
#24872 opened Nov 13, 2024 by chrisrothUT
AOT errors when setting compile options bug Something isn't working
#24869 opened Nov 12, 2024 by man2machine
CUDA_ERROR_SYSTEM_NOT_READY: system not yet initialized bug Something isn't working
#24866 opened Nov 12, 2024 by carlosgmartin
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
Bug in the latest libtpu nightly release bug Something isn't working
#24829 opened Nov 11, 2024 by knyazer
potential jax 0.4.35 release issue? bug Something isn't working
#24826 opened Nov 10, 2024 by haohuanw
Flaky test tests/fft_test.py::FftTest::testFftfreq5 bug Something isn't working
#24798 opened Nov 8, 2024 by apivovarov
Nightly builds on Mac ARM fail complex function numerical tests bug Something isn't working
#24787 opened Nov 8, 2024 by hawkinsp
shard_map doesn't work with jnp.insert bug Something isn't working
#24762 opened Nov 7, 2024 by mrlazy1708
pure_callback using a threadpool to accelerate vmap enhancement New feature or request
#24756 opened Nov 7, 2024 by Joshuaalbert
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
TPU hangs if using random inside a pure_callback call bug Something isn't working
#24720 opened Nov 5, 2024 by tengomucho
ProTip! Adding no:label will show everything without a label.