Skip to content

Pull requests: jax-ml/jax

New pull request New
563 Open 21,862 Closed
563 Open 21,862 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

Add ClosedJaxpr.{const,in,out}vars forwarding properties.
#16446 opened Jun 16, 2023 by dan-zheng Loading…
@mattjj
4
Expose unroll option supported in lax.scan into lax.map
#16341 opened Jun 10, 2023 by JoeyTeng Loading…
@hawkinsp
wip exploring cache explanations pull ready Ready for copybara import and testing
#16338 opened Jun 9, 2023 by mattjj Loading…
@mattjj
Update conv_transpose docstring to inform users about swapped spatial axes in kernel as compared to Keras
#16238 opened Jun 2, 2023 by SriRangaTarun Loading…
1
fix Dropout's bug
#16229 opened Jun 2, 2023 by fmscole Loading…
@froystig
5
improve a jaxpr tracing error message
#16090 opened May 23, 2023 by mattjj • Draft
@mattjj
in einsum, don't promote operand dtypes if preferred_element_type is given
#16087 opened May 22, 2023 by mattjj Loading…
@mattjj
2
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
Adds warning, in particular to catch Partial(Partial(...), ...)
#15999 opened May 14, 2023 by patrick-kidger Loading…
1
@hawkinsp
[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…
4
Checkify: instrumentation context manager
#15935 opened May 9, 2023 by LenaMartens • Draft
@LenaMartens
Update the HloModule generation instructions to use jax_to_ir.py
#15821 opened May 2, 2023 by LaurentMazare Loading…
Update the jaxlib license to reflect the current dependencies. pull ready Ready for copybara import and testing
#15777 opened Apr 27, 2023 by hawkinsp Loading…
@hawkinsp
Simplify JVP of lax.div by reusing already computed values
#15766 opened Apr 26, 2023 by lgeiger Loading…
@mattjj
4
add an api_boundary around device_put pull ready Ready for copybara import and testing
#15595 opened Apr 13, 2023 by mattjj Loading…
@mattjj
5
Checkify: instrumentation context manager
#15389 opened Apr 4, 2023 by LenaMartens • Draft
11
Implementation of scipy.ndimage.gaussian_filter1d and gaussian_filter
#15359 opened Apr 2, 2023 by renecotyfanboy Loading…
1
Added working example of forward-forward algorithm
#15348 opened Mar 31, 2023 by sowmen Loading…
Use modern packaging standards for setuptools
#15280 opened Mar 29, 2023 by eli-schwartz Loading…
5
hilbert transform similar to that in osp.signal pull ready Ready for copybara import and testing
#15211 opened Mar 24, 2023 by joglekara Loading…
12
jax.random: clean up handling of dtypes pull ready Ready for copybara import and testing
#15177 opened Mar 23, 2023 by jakevdp Loading…
@jakevdp
1
revive shapecheck, better than ever
#14842 opened Mar 8, 2023 by mattjj • Draft
@mattjj
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…
@froystig
1
improve vmap error messages with pytree info
#14748 opened Mar 2, 2023 by mattjj • Draft
1
@mattjj
[typing] Use ParamSpec in JIT annotation pull ready Ready for copybara import and testing
#14688 opened Feb 26, 2023 by NeilGirdhar Loading…
@jakevdp
47
ProTip! Follow long discussions with comments:>50.