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,317 Open 4,162 Closed
1,317 Open 4,162 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

Operator jax.lax.gather consumes a lot of memory on CPU bug Something isn't working
#23457 opened Sep 5, 2024 by daskol
6
Error exporting np.where when unrelated dimension variable is 0 bug Something isn't working
#23456 opened Sep 5, 2024 by tchatow
@gnecula
2
Multi-process GPU jobs fail on Slurm bug Something isn't working
#23452 opened Sep 5, 2024 by ChenAo-Phys
6
TPU operations don't appear if profile is too long bug Something isn't working
#23446 opened Sep 5, 2024 by jlin816
ValueError: Apply must be called with an RNG as the third argument, the required signature is: apply(params, state, rng, *a, **k). The object was of type <class 'jax._src.interpreters.partial_eval.DynamicJaxprTracer'>: Traced<ShapedArray(uint32[2])>with<DynamicJaxprTrace(level=1/0) bug Something isn't working
#23428 opened Sep 4, 2024 by eaven123
@jakevdp
5
Profile shows TPU operations before CPU instructions bug Something isn't working
#23426 opened Sep 4, 2024 by jlin816
jnp.arange does not permit dynamic shaped arrays bug Something isn't working
#23423 opened Sep 3, 2024 by josh146
@mattjj
2
CuSolver: Switch to 64 bit api to allow for eigh on matrices > than 26732x26732 enhancement New feature or request
#23413 opened Sep 3, 2024 by PhilipVinc
@dfm
6
Improve CuSolver errors diagnostics enhancement New feature or request
#23410 opened Sep 3, 2024 by PhilipVinc
@dfm
8
_isolve tacitly checks for symmetric linear operator wrongly bug Something isn't working
#23403 opened Sep 3, 2024 by Joshuaalbert
3
error: failed to legalize operation 'mhlo.erf' with jax-metal bug Something isn't working
#23384 opened Sep 2, 2024 by a-gn
@kulinseth @shuhand0
Segmentation Fault on 8x8 or larger matrix multiplication with GPU bug Something isn't working
#23355 opened Aug 30, 2024 by treyra
1
jax.nn.dot_product_attention does not respect key_value_seq_lengths bug Something isn't working
#23349 opened Aug 30, 2024 by danjenson
4
jnp.pow() triggers dtype promotion when it should not bug Something isn't working
#23337 opened Aug 30, 2024 by ayaka14732
3
Add function that checks whether jax is yet initialized or not enhancement New feature or request
#23320 opened Aug 29, 2024 by milutter
Full zero arrays allocated for "consumed" gradients of stop_gradient parameters bug Something isn't working
#23316 opened Aug 29, 2024 by colehaus
2
Cannot pass constant to lax.fori_loop body inside pallas kernel bug Something isn't working
#23301 opened Aug 28, 2024 by lengstrom
@justinjfu
1
Flash attention soft capping support enhancement New feature or request pallas Issues pertaining to Pallas (GPU or TPU)
#23300 opened Aug 28, 2024 by sparsh35
jax.numpy.tan call adds chlo.tan call to exported StableHLO; stablehlo.tan should be preferred bug Something isn't working
#23259 opened Aug 26, 2024 by joaospinto
1
4
sparse.BCOO error when matrix is passed as a parameter bug Something isn't working duplicate This issue or pull request already exists
#23237 opened Aug 26, 2024 by muchanem
@jakevdp
2
Wrong derivatives when using defjvp for sparse.BCOO matrices bug Something isn't working
#23235 opened Aug 26, 2024 by BodeTobias
2
[Important] Keep non-hermetic CUDA toolkit as a build option. bug Something isn't working
#23231 opened Aug 26, 2024 by yliu120
2
JITing loss function causes erroneous results bug Something isn't working
#23226 opened Aug 24, 2024 by muchanem
@pschuh
11
Guard against unintentional transfers to CPU enhancement New feature or request
#23215 opened Aug 23, 2024 by kxs-dhenshall
4
Rotation.concatenate does not work for two single rotations bug Something isn't working
#23202 opened Aug 22, 2024 by thomas-rkk
@jakevdp
1
ProTip! Mix and match filters to narrow down what you’re looking for.