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

Added butterworth filter code
#26824 opened Feb 27, 2025 by ParamThakkar123
Reimplement custom_vjp.optimize_remat using custom_dce pull ready Ready for copybara import and testing
#26814 opened Feb 27, 2025 by dfm Approved
Add a profiler test for gpu run
#26786 opened Feb 26, 2025 by copybara-service bot
fix Initializer protocol
#26765 opened Feb 26, 2025 by Qwlouse
Debug failing CUDA test jobs CI Connection Halt - On Retry Flag every job that has its workflow file setup for halting to halt if the workflow is retried
#26749 opened Feb 25, 2025 by copybara-service bot
Use the same name for aliased Vars when pretty-printing Jaxprs. pull ready Ready for copybara import and testing
#26697 opened Feb 24, 2025 by gnecula
ProTip! Filter pull requests by the default branch with base:main.