Skip to content

Pull requests: jax-ml/jax

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
Assignee
Filter by who’s assigned
Sort

Pull requests list

Test lax_scipy_spectral_dac error message
#25073 opened Nov 23, 2024 by copybara-service bot Loading…
[mgpu] Pointwise min
#25061 opened Nov 22, 2024 by copybara-service bot Loading…
[mgpu/pallas_mgpu] Pointwise sign function
#25060 opened Nov 22, 2024 by copybara-service bot Loading…
[mgpu/pallas_mgpu] Pointwise tanh support
#25059 opened Nov 22, 2024 by copybara-service bot Loading…
Use optimize='auto' for jnp.linalg.multi_dot
#25055 opened Nov 22, 2024 by dfm Loading…
Rework custom hermetic python instructions.
#25052 opened Nov 22, 2024 by copybara-service bot Loading…
Add option of disabling cudnn version checking.
#25044 opened Nov 21, 2024 by copybara-service bot Loading…
Reverts 525b646c0ebd5205f4fa0639c94adb2de47e1cf0
#25036 opened Nov 21, 2024 by copybara-service bot Loading…
[mgpu] Causal mask operator.
#25028 opened Nov 21, 2024 by copybara-service bot Loading…
allow custom_gradient to close over non-autodiff tracers pull ready Ready for copybara import and testing
#25017 opened Nov 20, 2024 by mattjj Loading…
[Mosaic GPU] Improve default kernel name and add option to customize pull ready Ready for copybara import and testing
#25006 opened Nov 20, 2024 by andportnoy Loading…
ProTip! Type g i on any issue or pull request to go back to the issue listing page.