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

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

custom_jvp leaks tracers if they're marked as nondiff_argnum better_errors Improve the error reporting
#23065 opened Aug 14, 2024 by jakevdp
@froystig @jakevdp
2
.at.set notation assigns the wrong values to the wrong array elements on a 2x2 TPU bug Something isn't working
#22827 opened Aug 1, 2024 by Sohl-Dickstein
@froystig
1
Support Hessian of gamma-distributed samples
#21432 opened May 25, 2024 by NeilGirdhar Loading…
1
@froystig
4
Numerical differences between shardings in random algorithm bug Something isn't working
#21232 opened May 14, 2024 by shawnwang18
@froystig
3
buggy interaction: remat, automatic partitioning, and unsafe rbg-based RNGs bug Something isn't working
#21046 opened May 3, 2024 by froystig
@froystig @mattjj
large groups of derived test cases are jointly sensitive to their common random seed bug Something isn't working
#18941 opened Dec 12, 2023 by froystig
@froystig @jakevdp
4
random_lax_test::LaxRandomTest.testBall is sensitive to its random seed bug Something isn't working
#18932 opened Dec 12, 2023 by froystig
@froystig @jakevdp
Large consume of memory when using jax.lax.cond bug Something isn't working
#18769 opened Dec 1, 2023 by celiolarcher
@froystig
1
abstract value protocol / ShapeDtypeStruct distinction cleanup Code cleanups and improvements enhancement New feature or request
#18459 opened Nov 9, 2023 by froystig
@froystig
Add example of jaxpr pretty_print(source_info=True) to the make_jaxpr documentation documentation enhancement New feature or request
#18240 opened Oct 23, 2023 by fcole
@froystig
give users a way to type-query effects, e.g. in eval_shape enhancement New feature or request
#18239 opened Oct 23, 2023 by mattjj
@froystig @sharadmv @mattjj
4
Gradient of jax.random.dirichlet with small alpha element gives nan bug Something isn't working
#18129 opened Oct 16, 2023 by hylkedonker
@froystig
4
jax_threefry_partitionable + rematerialization doesn't seem to be working together in distributed training bug Something isn't working
#17982 opened Oct 6, 2023 by hr0nix
@froystig
16
Sharp edge in (probably ill-advised) custom_vmap usecase bug Something isn't working
#17342 opened Aug 29, 2023 by sschoenholz
@froystig
accept longer seeds when initializing RNG keys enhancement New feature or request
#16808 opened Jul 20, 2023 by froystig
@froystig
document class methods in autodiff autosummary pull ready Ready for copybara import and testing
#16639 opened Jul 6, 2023 by froystig Loading…
@froystig
1
fix Dropout's bug
#16229 opened Jun 2, 2023 by fmscole Loading…
@froystig
5
Stax has a bug, Dropout cannot convert from training mode to test mode bug Something isn't working
#16218 opened Jun 1, 2023 by fmscole
@froystig @mattjj
1
support all rng_bit_generator algorithms as RNG implementations pull ready Ready for copybara import and testing
#16034 opened May 17, 2023 by froystig Loading…
@froystig
2
jax.custom_jvp promotes tangent-of-integer to integer (not float0) bug Something isn't working
#16000 opened May 14, 2023 by patrick-kidger
@froystig @mattjj @LenaMartens
1
RNG slows down data parallel training bug Something isn't working performance make things lean and fast
#15895 opened May 5, 2023 by froystig
@froystig
6
extensions module enhancement New feature or request JEP JAX enhancement proposal
#15856 opened May 3, 2023 by froystig
@froystig @jakevdp
@sharadmv @yashk2810
3
"No constant handler" error appearing after jit/pjit merge bug Something isn't working
#15759 opened Apr 26, 2023 by DanPuzzuoli
@froystig @mattjj
4
consume (at least) 64-bit messages in fold_in enhancement New feature or request
#15296 opened Mar 29, 2023 by froystig
@froystig
avoid eagerly staging custom_vmap'ed functions enhancement New feature or request
#14961 opened Mar 13, 2023 by Edenhofer
@froystig
2
ProTip! Updated in the last three days: updated:>2024-09-10.