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

[vmappable] fix trace context bugs kokoro:force-run pull ready Ready for copybara import and testing
#27759 opened Apr 6, 2025 by mattjj Loading…
Address previous FP8-related TODOs in jaxlib/XLA.
#27757 opened Apr 5, 2025 by copybara-service bot Loading…
Reverts e2f67e0ef1af19f4d32e02f6fb927502469b32c6
#27752 opened Apr 4, 2025 by copybara-service bot Loading…
roll back test-breaking change
#27751 opened Apr 4, 2025 by copybara-service bot Loading…
Set varying_axes_in_types to True
#27749 opened Apr 4, 2025 by copybara-service bot Loading…
Create a PjRt protos dir.
#27744 opened Apr 4, 2025 by copybara-service bot Loading…
Fix jax.random.randint for 8-bit and 16-bit dtypes kokoro:force-run pull ready Ready for copybara import and testing
#27742 opened Apr 4, 2025 by jakevdp Loading…
Test download time.
#27741 opened Apr 4, 2025 by copybara-service bot Loading…
Apply forwarding in pjit linearization rule to avoid intermediate copies kokoro:force-run pull ready Ready for copybara import and testing
#27735 opened Apr 4, 2025 by dfm Loading…
Add jax.fwd_and_bwd
#27733 opened Apr 4, 2025 by gspschmid Loading…
Migrate jaxlib to use a single common .so file for all C++ dependencies. kokoro:force-run pull ready Ready for copybara import and testing windows:force-run Apply this label to a PR to force a run of the windows CI
#27732 opened Apr 4, 2025 by hawkinsp Loading…
Plumbing for has-a between LoadedExec and Exec
#27727 opened Apr 4, 2025 by copybara-service bot Loading…
Fix shard_map for primitives with multiple outputs and None specs kokoro:force-run pull ready Ready for copybara import and testing
#27720 opened Apr 3, 2025 by justinjfu Loading…
jax.numpy: support __jax_array__ in remaining APIs kokoro:force-run pull ready Ready for copybara import and testing
#27716 opened Apr 3, 2025 by jakevdp Loading…
Move DropVar annotation into pretty printer pull ready Ready for copybara import and testing
#27709 opened Apr 3, 2025 by dfm Loading…
ProTip! Adding no:label will show everything without a label.