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

Clear current search query, filters, and sorts
6 Open 81 Closed
6 Open 81 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

Add missing test for input shardings being None and output
#23286 opened Aug 28, 2024 by copybara-service bot Loading…
@pschuh
JITing loss function causes erroneous results bug Something isn't working
#23226 opened Aug 24, 2024 by muchanem
@pschuh
12
Errors from interaction between shard_map with auto and lax.map bug Something isn't working
#23019 opened Aug 12, 2024 by jeffgortmaker
@pschuh
Keyboard interrupt not working bug Something isn't working
#18246 opened Oct 23, 2023 by danijar
@pschuh
5
Document spmd_axis_name in jax.vmap enhancement New feature or request
#15889 opened May 5, 2023 by shoyer
@pschuh
2
using pmap causing slowly gpu memory increases and eventually OOM in the end bug Something isn't working
#11357 opened Jul 3, 2022 by WingsOfPanda
@pschuh
3
ProTip! Exclude everything labeled bug with -label:bug.