Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
268 Open 8,128 Closed
268 Open 8,128 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

fix(doc): no autoselect of attention implementation
#32199 opened Sep 30, 2025 by emergenz Loading…
Remove unused tf_vendored repository rule.
#32198 opened Sep 30, 2025 by copybara-service bot Loading…
Automated Code Change
#32197 opened Sep 30, 2025 by copybara-service bot Loading…
Bump hypothesis from 6.138.16 to 6.140.2 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#32188 opened Sep 29, 2025 by dependabot bot Loading…
[Mosaic GPU] Saturate Divides constraints for equal vars.
#32185 opened Sep 29, 2025 by copybara-service bot Loading…
[Mosaic GPU] Canonicalize dimensions_to_tile in the Divides constraint.
#32182 opened Sep 29, 2025 by copybara-service bot Loading…
[Mosaic GPU] Add a helper method to check if an ir value is an SMEM reference.
#32178 opened Sep 29, 2025 by copybara-service bot Loading…
[Mosaic GPU] Add enable_smem_inference to DerivationContext.
#32176 opened Sep 29, 2025 by copybara-service bot Loading…
[Mosaic GPU] Add code that will assign SMEM transforms once they are inferred.
#32175 opened Sep 29, 2025 by copybara-service bot Loading…
[Mosaic GPU] Add a helper method to compute the swizzle given a type and a tiling.
#32173 opened Sep 29, 2025 by copybara-service bot Loading…
Use triple overloads of TargetRegistry functions
#32161 opened Sep 28, 2025 by boomanaiden154 Loading…
add aval_property.setter
#32151 opened Sep 26, 2025 by cgarciae Loading…
Remove notap flag from B200 pallas tests
#32141 opened Sep 26, 2025 by copybara-service bot Loading…
[Mosaic GPU] Add a Divides constraint type.
#32131 opened Sep 26, 2025 by copybara-service bot Loading…
[JAX] Add compiled program call_location logging support to JAX
#32124 opened Sep 25, 2025 by copybara-service bot Loading…
[Mosaic] Expand arith.select canonicalization
#32122 opened Sep 25, 2025 by copybara-service bot Loading…
[CI] Ignore: Tune the B200 tests per accelerator CI Connection Halt - On Retry Flag every job that has its workflow file setup for halting to halt if the workflow is retried
#32121 opened Sep 25, 2025 by MichaelHudgins • Draft
[Mosaic TPU] Support scalar input-output-alias. Relax the condition in ShapeVerifier::HandleCustomCall to allow scalar operand live on S(6) while the result has no memory space assigned.
#32116 opened Sep 25, 2025 by copybara-service bot Loading…
Relax version requirements for ROCm Jax Plugin wheels kokoro:force-run pull ready Ready for copybara import and testing
#32115 opened Sep 25, 2025 by JehandadKhan Loading…
2
Update rules_ml_toolchain to a newer version.
#32110 opened Sep 25, 2025 by copybara-service bot Loading…
Change tf_vendored repo macro to a local_repository wrapper and remove it where it's unused.
#32096 opened Sep 25, 2025 by copybara-service bot Loading…
Reverts 4d8b701068d25c84c123481751a0ddc3be08d6ab
#32093 opened Sep 25, 2025 by copybara-service bot Loading…
[Mosaic:TPU][apply-vector-layout] tpu.concatenate for result offsets != 0
#32091 opened Sep 25, 2025 by copybara-service bot Loading…
[Mosaic:TPU][infer-vector-layout] Fix bug in tpu.concatenate rule where invalid offsets can be assigned
#32090 opened Sep 25, 2025 by copybara-service bot Loading…
[Mosaic:TPU] Bitcast vreg instead of mask in SelectWithBounds
#32087 opened Sep 24, 2025 by copybara-service bot Loading…
ProTip! Adding no:label will show everything without a label.