-
Notifications
You must be signed in to change notification settings - Fork 3.3k
Pull requests: jax-ml/jax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Add LoadedExecutable.serialize() to match Executable.serialize().
#33786
opened Dec 6, 2025 by
copybara-service
bot
Loading…
improve tracer reprs to show a lil concrete info
kokoro:force-run
pull ready
Ready for copybara import and testing
#33785
opened Dec 6, 2025 by
mattjj
Loading…
Allow overriding the XLA commit to use in Continuous workflow
#33784
opened Dec 6, 2025 by
copybara-service
bot
Loading…
[jax.collect_profile] Allow arbitrary options to be passed to XProf
#33782
opened Dec 5, 2025 by
copybara-service
bot
Loading…
[Pallas TPU] Allow closed over scalars in core_map code
#33781
opened Dec 5, 2025 by
copybara-service
bot
Loading…
Add collective to async_copy_scales_to_tmem.
#33780
opened Dec 5, 2025 by
copybara-service
bot
Loading…
[Pallas SC] Allow semaphores to be returned by SCS kernels
#33779
opened Dec 5, 2025 by
copybara-service
bot
Loading…
[pallas:mosaic]
pltpu.emit_pipeline now accepts block specs in HBM
#33776
opened Dec 5, 2025 by
copybara-service
bot
Loading…
Change PjRt to use new copy of coordination service.
#33775
opened Dec 5, 2025 by
copybara-service
bot
Loading…
Move rules_ml_toolchain up due to request move default rules_ml_toolchain initialization to @xla//:workspace4.bzl
#33774
opened Dec 5, 2025 by
copybara-service
bot
Loading…
Add initial blackwell mxfp4 blockscale matmul implementation.
#33770
opened Dec 5, 2025 by
copybara-service
bot
Loading…
[pallas:mosaic] Access memory spaces via
pltpu directly
#33768
opened Dec 5, 2025 by
copybara-service
bot
Loading…
[pallas:sc]
pltpu.emit_pipeline now supports 1D-tiled block specs
#33767
opened Dec 5, 2025 by
copybara-service
bot
Loading…
[mosaic] Added documentation and a few useful methods to tpu.tiled attribute
#33763
opened Dec 5, 2025 by
copybara-service
bot
Loading…
Adjust tile_n assert to accommodate 2 cta in tcgen05 blockscale mma.
#33760
opened Dec 5, 2025 by
copybara-service
bot
Loading…
[export] Add backwards compatibility tests for serialized exports
pull ready
Ready for copybara import and testing
#33758
opened Dec 5, 2025 by
gnecula
Loading…
[Mosaic GPU] Add support for a limited class of reshapes for tiled layouts
#33756
opened Dec 5, 2025 by
copybara-service
bot
Loading…
Merge repr and str of array in one function instead of duplicating conditions
#33750
opened Dec 5, 2025 by
copybara-service
bot
Loading…
Previous Next
ProTip!
Add no:assignee to see everything that’s not assigned.