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
Label
Projects
Milestones
Assignee
Sort
Dot of dn=((0,),(0,)) for a=[100, 100] b=[100, 1] crashes on GPU with latest JAX release
bug
#7752
opened Aug 29, 2021 by
tomhennigan
Can't build jax 0.1.60 on mac 10.13.6 with CUDA10.1 + cuDNN7.6.5
bug
build
#7741
opened Aug 27, 2021 by
llv22
Jaxlib : Compiling/Building from source failed (Linux, cuda enabled)
bug
#7712
opened Aug 25, 2021 by
Corentin-pro
hoist named shape logic from threefry's random_bits implementation
enhancement
P2 (eventual)
#7709
opened Aug 24, 2021 by
froystig
Split keys over named batch axis
enhancement
P2 (eventual)
#7708
opened Aug 24, 2021 by
dieterichlawson
make pmap error with unhashable values as static_broadcasted_argnums
bug
#7703
opened Aug 24, 2021 by
mattjj
testSphHarmOrderZeroDegreeOne() in lax_scipy_test.py is not working for GPU runtime.
bug
#7700
opened Aug 24, 2021 by
reza-amd
[jax2tf] Converted functions should respect jax.jit, and force TF jit
enhancement
P1 (soon)
#7677
opened Aug 20, 2021 by
gnecula
JAX should not create duplicate HLO computations for reductions
enhancement
#7654
opened Aug 18, 2021 by
cheshire
RET_CHECK failure hlo->has_sharding() Side-effect HLO must have sharding
bug
#7635
opened Aug 16, 2021 by
sudhakarsingh27
Trying to get a PRNGKey on cpu results in strange gpu memory allocation
question
#7622
opened Aug 13, 2021 by
ysngshn
Previous Next
ProTip!
no:milestone will show everything without a milestone.

