Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
74 Open 6,771 Closed
74 Open 6,771 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
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Fix init step sizes to account for the time span of integration cla: yes
#3442 opened Jun 15, 2020 by samuela Loading… updated Jun 17, 2021
1
@shoyer
11
Clarify error message in scan transpose cla: yes
#3917 opened Jul 30, 2020 by jekbradbury Loading… updated Jun 17, 2021
2
Remove TraceContext & StepTraceContext cla: yes
#7050 opened Jun 22, 2021 by kaushikb11 Loading… updated Jun 29, 2021
@skye
8
Improve error message when passing a tuple of non-concrete values as … better_errors Improve the error reporting cla: yes pull ready Ready for copybara import and testing
#7952 opened Sep 17, 2021 by hawkinsp Loading… updated Sep 20, 2021
@hawkinsp
add gpu determinism note cla: yes
#4824 opened Nov 7, 2020 by mattjj Loading… updated Nov 11, 2021
@mattjj
6
fix jnp.convolve int precision pull ready Ready for copybara import and testing
#9953 opened Mar 18, 2022 by mattjj Loading… updated Mar 30, 2022
@mattjj
1
add is_leaf to tree_util.all_leaves pull ready Ready for copybara import and testing
#10213 opened Apr 9, 2022 by mattjj Loading… updated Apr 11, 2022
@mattjj
3
improve custom_jvp nondiff_argnums tracer error kokoro:force-run pull ready Ready for copybara import and testing
#12495 opened Sep 23, 2022 by mattjj Loading… updated Sep 23, 2022
Add PYTORCH_SAME padding to JAX. pull ready Ready for copybara import and testing
#12659 opened Oct 4, 2022 by RerRayne Loading… updated Oct 17, 2022
@apaszke
21
add scipy.special.factorial and scipy.special.comb pull ready Ready for copybara import and testing
#10948 opened Jun 2, 2022 by froystig • Draft updated Nov 15, 2022
@froystig
4
fix broken notebook link pull ready Ready for copybara import and testing
#13607 opened Dec 11, 2022 by mattjj Loading… updated Dec 11, 2022
@mattjj
5
Fix wrong cond-state discharge rule. pull ready Ready for copybara import and testing
#13785 opened Dec 24, 2022 by patrick-kidger Loading… updated Jan 11, 2023
@sharadmv
1
compute jit number of replicas using least-common-multiple, not max kokoro:force-run pull ready Ready for copybara import and testing
#13938 opened Jan 10, 2023 by mattjj Loading… updated Jan 13, 2023
1
@mattjj
Added xla_call and custom_jvp_call discharge rules pull ready Ready for copybara import and testing
#13837 opened Dec 31, 2022 by patrick-kidger Loading… updated Jan 13, 2023
@sharadmv
jax.random: clean up handling of dtypes pull ready Ready for copybara import and testing
#15177 opened Mar 23, 2023 by jakevdp Loading… updated Mar 23, 2023
@jakevdp
1
hilbert transform similar to that in osp.signal pull ready Ready for copybara import and testing
#15211 opened Mar 24, 2023 by joglekara Loading… updated Mar 28, 2023
12
in closure_convert, hoist all tracers involved in staging pull ready Ready for copybara import and testing
#14760 opened Mar 2, 2023 by froystig Loading… updated May 15, 2023
@froystig
1
[ANN] Update approx_{max,min}_k jvp to make it more TPU-friendly. pull ready Ready for copybara import and testing
#15977 opened May 11, 2023 by ebrevdo Loading… updated May 15, 2023
4
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… updated May 17, 2023
@froystig
2
add an api_boundary around device_put pull ready Ready for copybara import and testing
#15595 opened Apr 13, 2023 by mattjj Loading… updated Jun 27, 2023
@mattjj
5
Enhance Consistency of Kaiming He Initializer Documentation pull ready Ready for copybara import and testing
#16998 opened Aug 7, 2023 by peterdavidfagan Loading… updated Aug 8, 2023
1
Added vectorized arguments to debug.{print,breakpoint,callback}. pull ready Ready for copybara import and testing
#17088 opened Aug 12, 2023 by patrick-kidger Loading… updated Sep 6, 2023
2
improve ravel_pytree docstring with an example, add type annotation pull ready Ready for copybara import and testing
#18283 opened Oct 26, 2023 by mattjj Loading… updated Nov 8, 2023
@mattjj
add option to cond to disable double-where batching behavior pull ready Ready for copybara import and testing
#16875 opened Jul 27, 2023 by mattjj Loading… updated Nov 9, 2023
@mattjj
improve error message for #14009 pull ready Ready for copybara import and testing
#16841 opened Jul 25, 2023 by mattjj Loading… updated Nov 9, 2023
1
@mattjj
ProTip! Mix and match filters to narrow down what you’re looking for.