Skip to content

Issues: google/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,291 Open 3,841 Closed
1,291 Open 3,841 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
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

ROCm 6.1, 7900 xtx: bfloat16 support not enabled? bug Something isn't working
#21074 opened May 4, 2024 by brettkoonce
Persistent compilation cache does not work bug Something isn't working
#21067 opened May 3, 2024 by neel04
2
Unexpected speedup from wrapping function call in trivial jax.lax.cond statement bug Something isn't working
#21065 opened May 3, 2024 by TonyZhou729
dynamic config scope under jit doesn't change partitionable threefry behavior bug Something isn't working
#21061 opened May 3, 2024 by froystig
@froystig
Seeking guidance for landing spot of scipy.stats.levy_stable in Jax enhancement New feature or request
#21054 opened May 3, 2024 by tjhunter
buggy interaction: remat, automatic partitioning, and unsafe rbg-based RNGs bug Something isn't working
#21046 opened May 3, 2024 by froystig
@froystig @mattjj
spsolve exits with error when inverting matrix sum bug Something isn't working
#21031 opened May 1, 2024 by buvoli
@jakevdp
4
jnp.fft.ifft imprecision for GPU bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#21002 opened Apr 30, 2024 by chih-kang-huang
⚠️ Nightly upstream-dev CI failed ⚠️ CI
#20996 opened Apr 30, 2024 by github-actions bot
Sharding is much slower than pmap for while loops of varying length while loops bug Something isn't working
#20968 opened Apr 27, 2024 by lockwo
2
Counterintuitive speed of einsums vs equivalent matmuls bug Something isn't working
#20952 opened Apr 26, 2024 by shehzaidi
5
Performance Issue Report: JAX Slower Than Autograd on GPU and CPU Setups bug Something isn't working type:performance
#20948 opened Apr 26, 2024 by alibastami
4
Metal : fp64 operations with jax.numpy base functions not supported Apple GPU (Metal) plugin bug Something isn't working
#20938 opened Apr 25, 2024 by aboucaud
@kulinseth @shuhand0
3
Adding tree_util.stack_leaves() and tree_util.unstack_leaves() enhancement New feature or request
#20934 opened Apr 25, 2024 by ayaka14732
5
sum(keepdims=True) should work in pallas bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#20921 opened Apr 24, 2024 by voznesenskym
@apaszke
jax.numpy.insert returning incorrect results wen jitted on Metal Apple GPU (Metal) plugin bug Something isn't working
#20918 opened Apr 24, 2024 by fkeruzore
@shuhand0
1
Pallas jax.lax.fori_loop over long inputs slows down bug Something isn't working
#20909 opened Apr 24, 2024 by jbuckman
1
Larger matrix diagonalization returns OverflowError enhancement New feature or request
#20904 opened Apr 23, 2024 by OMalenfantThuot
@hawkinsp
Pallas strided loads ignore the stride argument on CUDA backend bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#20895 opened Apr 23, 2024 by hirayaku
improve error message and docs for custom_jvp / custom_vjp nondiff_argnums escaped tracer error better_errors Improve the error reporting documentation
#20889 opened Apr 23, 2024 by mattjj
@mattjj
Better Handling of Nested Loop with Remat enhancement New feature or request
#20877 opened Apr 23, 2024 by LeoXinhaoLee
Unbound axis names within scan within custom partitioning bug Something isn't working
#20864 opened Apr 22, 2024 by nshepperd
@superbobry
jax.nn.softmax is inconsistent under jax.jit bug Something isn't working
#20856 opened Apr 21, 2024 by francois-rozet
@jakevdp
7
Build fails due to /usr/bin/ccache being detected as Clang by build/build.py bug Something isn't working
#20841 opened Apr 19, 2024 by andportnoy
Probability density/mass functions from jax.scipy.stats not supported on Metal Apple GPU (Metal) plugin bug Something isn't working
#20835 opened Apr 19, 2024 by logan-blaine
@kulinseth @shuhand0
1
ProTip! Find all open issues with in progress development work with linked:pr.