Skip to content

Pull requests: google/jax

New pull request New
401 Open 14,793 Closed
401 Open 14,793 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Adding JAX_LOGGING_LEVEL configuration option
#23468 opened Sep 6, 2024 by rdyro Loading…
Revert to nvtx_kern_sum to get correct profiling info and adding post processing logic pull ready Ready for copybara import and testing
#23467 opened Sep 5, 2024 by abhinavgoel95 Loading…
2
Fix triton capi_objects target to depend on MLIR CAPIIRObjects bazel pull ready Ready for copybara import and testing
#23462 opened Sep 5, 2024 by hawkinsp Loading…
3
Improve docs for jnp.split & related APIs pull ready Ready for copybara import and testing
#23459 opened Sep 5, 2024 by jakevdp Loading…
@jakevdp
1
Fix expression parentheses in shape polymorphism docs for division of symbolic dimensions. pull ready Ready for copybara import and testing
#23454 opened Sep 5, 2024 by carlosgmartin Loading…
A few more pyright fixes
#23453 opened Sep 5, 2024 by superbobry Loading…
3
Added shape and dtype to AbstractValue
#23449 opened Sep 5, 2024 by superbobry Loading…
2
Activate Tridiagonal Reduction to XLA's FFI
#23447 opened Sep 5, 2024 by copybara-service bot Loading…
1
DOC: Improved documentation for jax.numpy.kron and jax.numpy.outer
#23443 opened Sep 4, 2024 by selamw1 Loading…
@jakevdp
10
Add iota_order=True for broken testcases.
#23438 opened Sep 4, 2024 by copybara-service bot Loading…
1
Added explicit overlap example to the shard_map tutorial
#23436 opened Sep 4, 2024 by chaserileyroberts Loading…
1
Skip tests for 64-bit types when jax_enable_x64 is set to false.
#23432 opened Sep 4, 2024 by copybara-service bot Loading…
1
doc: update examples of deprecation timelines documentation pull ready Ready for copybara import and testing
#23424 opened Sep 3, 2024 by jakevdp Loading…
@jakevdp
Add frontend attributes to Jax. This allows Jax users to annotate Jax code with frontend_attributes which can be traced down to the HLO level, to be used for numerical debugging purposes.
#23421 opened Sep 3, 2024 by copybara-service bot Loading…
[NVIDIA] Relax the requirement for providing both query_seq_lengths and key_value_seq_lengths
#23415 opened Sep 3, 2024 by kaixih Loading…
6
Delete all redundant environment variables from RBE configs.
#23414 opened Sep 3, 2024 by copybara-service bot Loading…
Enable the enhanced TPU launch barrier on all TPU generations.
#23411 opened Sep 3, 2024 by copybara-service bot Loading…
@hawkinsp
Add FFI example project and test on CI pull ready Ready for copybara import and testing
#23409 opened Sep 3, 2024 by dfm Loading…
@dfm
7
CHLO -> StableHLO : use TanOp and StablehloCreateCompatibilityExpanderPass
#23408 opened Sep 3, 2024 by copybara-service bot Loading…
[Pallas] Fix integer array indexing pallas Issues pertaining to Pallas (GPU or TPU)
#23406 opened Sep 3, 2024 by ayaka14732 • Draft
1
@ayaka14732
Update ravel_multi_index to accept non-concrete values.
#23391 opened Sep 3, 2024 by copybara-service bot Loading…
Bump actions/upload-artifact from 4.3.6 to 4.4.0 dependencies Pull requests that update a dependency file github_actions Pull requests that update Github_actions code
#23383 opened Sep 2, 2024 by dependabot bot Loading…
Bump actions/setup-python from 5.1.1 to 5.2.0 dependencies Pull requests that update a dependency file github_actions Pull requests that update Github_actions code
#23382 opened Sep 2, 2024 by dependabot bot Loading…
Bump contourpy from 1.2.1 to 1.3.0 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#23381 opened Sep 2, 2024 by dependabot bot Loading…
Bump rich from 13.7.1 to 13.8.0 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#23380 opened Sep 2, 2024 by dependabot bot Loading…
ProTip! Follow long discussions with comments:>50.