Skip to content

Pull requests: google-deepmind/tf2jax

New pull request New
17 Open 225 Closed
17 Open 225 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

Finalize deprecation of xla_extension.
#249 opened Jul 22, 2025 by copybara-service bot Loading…
Accelerate deprecation for attributes in jax.lib.xla_client and redirect dependencies to xla/python/xla_client.
#247 opened Jul 16, 2025 by copybara-service bot Loading…
Improve Minor Errors for Clarity
#242 opened May 9, 2025 by yus100 Loading…
[JAX] Deprecate jax.lib.xla_client.{heap_profile,mlir_api_version}.
#234 opened Apr 9, 2025 by copybara-service bot Loading…
Update references to JAX's GitHub repo
#227 opened Dec 5, 2024 by copybara-service bot Loading…
feat: add xlogy op
#219 opened Oct 3, 2024 by s22chan Loading…
1
fix(ops): sizes == -1 was interpreted incorrectly
#218 opened Oct 3, 2024 by s22chan Loading…
2
[JAX] add support for gather/scatter batching dims following the new attributes in stablehlo.
#214 opened Sep 21, 2024 by copybara-service bot Loading…
OOB whatever; this CL description should be cleaned up before sent for review.
#194 opened May 28, 2024 by copybara-service bot Loading…
1
Implement "bad_indcies_policy" for ScatterNd.
#193 opened May 27, 2024 by copybara-service bot Loading…
1
ScatterNdFunctor to update operands of all valid indices and continue on bad indices.
#192 opened May 27, 2024 by copybara-service bot Loading…
1
Add "bad_indcies_policy" to ScatterNd
#191 opened May 27, 2024 by copybara-service bot Loading…
1
Reenable pytype in test.sh
#187 opened Apr 6, 2024 by copybara-service bot Loading…
Bump tensorflow from 2.11.0 to 2.11.1 dependencies Pull requests that update a dependency file
#176 opened Dec 22, 2023 by dependabot bot Loading…
Sort ops.
#166 opened Sep 28, 2023 by copybara-service bot Loading…
Use lax.cond for subsampled agent.
#152 opened Aug 9, 2023 by copybara-service bot Loading…
1
Temporary comment since I have tf2jax library changes on the diffbase.
#151 opened Aug 9, 2023 by copybara-service bot Loading…
1
ProTip! no:milestone will show everything without a milestone.