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,289 Open 3,831 Closed
1,289 Open 3,831 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

amd-smi import error with rocm 6.1 + rocm/jax:latest image bug Something isn't working
#20955 opened Apr 26, 2024 by brettkoonce
Counterintuitive speed of einsums vs equivalent matmuls bug Something isn't working
#20952 opened Apr 26, 2024 by shehzaidi
4
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
1
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
"pure_callback" is not exported from module "jax" bug Something isn't working
#20925 opened Apr 24, 2024 by carlosgmartin
1
@jakevdp
4
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
⚠️ Nightly upstream-dev CI failed ⚠️ CI
#20913 opened Apr 24, 2024 by github-actions bot
@jakevdp
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
Make custom VJP bwd shape/type checking optional. enhancement New feature or request
#20827 opened Apr 19, 2024 by davidhchoi
An uninformative error on TPU bug Something isn't working
#20812 opened Apr 18, 2024 by sash-a
2
Print large value with suppress in printoptions bug Something isn't working
#20807 opened Apr 18, 2024 by sh0416
1
Wrong output of jax.scipy.special.sph_harm bug Something isn't working
#20769 opened Apr 16, 2024 by SGENZO
1
3
Latency Hiding Scheduler leads to x5 memory usage if used without jax.lax.scan bug Something isn't working
#20763 opened Apr 15, 2024 by qGentry
jax.lax.scan() segmentation fault with jax-metal on Mac M1 Apple GPU (Metal) plugin bug Something isn't working
#20750 opened Apr 14, 2024 by kdesoto-astro
@shuhand0
1
[README] nn libraries fix documentation
#20745 opened Apr 14, 2024 by keren9898
26
ProTip! Updated in the last three days: updated:>2024-04-24.