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.

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,554 Open 4,556 Closed
1,554 Open 4,556 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

Different checkpoint behaviors in different version of JAX and CUDA bug Something isn't working
#27748 opened Apr 4, 2025 by Honghui-Du
*RuntimeError: bad conversion* when trying to register custom jax plugin bug Something isn't working
#27743 opened Apr 4, 2025 by kmitrovicTT
Profile whats inside jax.lax.fori_loop enhancement New feature or request
#27705 opened Apr 3, 2025 by cdelv
randint() produces non-uniform distribution for uint8 bug Something isn't working
#27702 opened Apr 3, 2025 by martinxyz
@jakevdp
4
Implement lax.scan and lax.while jet_rules for Taylor series propagation in jax.experimental.jet #26801 enhancement New feature or request
#27699 opened Apr 3, 2025 by DavidLanders95
@mattjj
Mutable Array type error bug Something isn't working skill issue
#27683 opened Apr 2, 2025 by cgarciae
2
@mattjj
3
JAX 0.4.38 breaks jax.numpy.split in shard_map bug Something isn't working
#27673 opened Apr 2, 2025 by jagourq
@justinjfu @yashk2810
2
pmin / shard_map - Casting of Integer Axis Names to Float bug Something isn't working
#27665 opened Apr 2, 2025 by leo1200
@yashk2810
2
shard_map - NotImplementedError: No replication rule for while bug Something isn't working
#27664 opened Apr 2, 2025 by leo1200
@mattjj
1
Add argument "largest" to top_k enhancement New feature or request
#27623 opened Mar 31, 2025 by carlosgmartin
@jakevdp
1
module 'jaxlib' has no attribute 'version' bug Something isn't working
#27614 opened Mar 31, 2025 by JujuCCP
3
Jitting a function with enable_x64 does not work for jnp.linalg.solve and related functions. bug Something isn't working
#27602 opened Mar 30, 2025 by mishavanbeek
@dfm @jakevdp
6
jax.lax.scan does not allow static loop variables for computing static values (e.g. for jnp.pad) enhancement New feature or request
#27601 opened Mar 30, 2025 by vietluong2110
5
jax.jit has general issues with gfx906 [ROCM] AMD GPU Issues pertaining to AMD GPUs (ROCM) bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#27600 opened Mar 30, 2025 by zmelumian972
2
Use MiniDebugInfo debug information on Linux wheel shared libraries to reduce wheel size build enhancement New feature or request type:others
#27598 opened Mar 30, 2025 by hawkinsp
2
Clarify whether lax.top_k returns results in sorted order documentation
#27594 opened Mar 30, 2025 by carlosgmartin
@jakevdp
4
jax.vmap and jax.lax.map result changes drastically with large array/batch_size bug Something isn't working
#27591 opened Mar 29, 2025 by gbuzzard
1
@dfm
6
Remove nanobind pin in FFI example requirements cleanup Code cleanups and improvements
#27563 opened Mar 28, 2025 by danielsuo
@danielsuo
jax.jit(..., inline=True) converts arrays back into Python scalars. bug Something isn't working
#27545 opened Mar 27, 2025 by patrick-kidger
1
Jax is unable to findn and uze the gpu bug Something isn't working needs info More information is required to diagnose & prioritize the issue.
#27535 opened Mar 27, 2025 by Its-a-me-Ashwin
1
[mosaic] Indexing into 3rd minor-most+ dimensionwith sub 32-bit dtypes enhancement New feature or request
#27532 opened Mar 27, 2025 by oliverdutton-iso
2
Lower Precision Grad Reduction bug Something isn't working
#27496 opened Mar 26, 2025 by jfc4050
@dfm @mattjj
2
Implementation of scipy.integrate.quad into jax.scipy enhancement New feature or request
#27493 opened Mar 26, 2025 by szoppelt
2 tasks done
@jakevdp
1
[Pallas] support negative indexing of array refs enhancement New feature or request
#27490 opened Mar 26, 2025 by ysngshn
4
error_check API does not support usage within vmap bug Something isn't working
#27483 opened Mar 26, 2025 by emergenz
@mattjj @dougalm @ayaka14732
ProTip! Follow long discussions with comments:>50.