-
Notifications
You must be signed in to change notification settings - Fork 2.7k
Pull requests: google/jax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
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…
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…
Improve docs for jnp.split & related APIs
pull ready
Ready for copybara import and testing
#23459
opened Sep 5, 2024 by
jakevdp
Loading…
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…
Activate Tridiagonal Reduction to XLA's FFI
#23447
opened Sep 5, 2024 by
copybara-service
bot
Loading…
DOC: Improved documentation for jax.numpy.kron and jax.numpy.outer
#23443
opened Sep 4, 2024 by
selamw1
Loading…
Add iota_order=True for broken testcases.
#23438
opened Sep 4, 2024 by
copybara-service
bot
Loading…
Added explicit overlap example to the shard_map tutorial
#23436
opened Sep 4, 2024 by
chaserileyroberts
Loading…
Skip tests for 64-bit types when
jax_enable_x64
is set to false.
#23432
opened Sep 4, 2024 by
copybara-service
bot
Loading…
doc: update examples of deprecation timelines
documentation
pull ready
Ready for copybara import and testing
#23424
opened Sep 3, 2024 by
jakevdp
Loading…
[NVIDIA] Relax the requirement for providing both
query_seq_lengths
and key_value_seq_lengths
#23415
opened Sep 3, 2024 by
kaixih
Loading…
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…
Add FFI example project and test on CI
pull ready
Ready for copybara import and testing
#23409
opened Sep 3, 2024 by
dfm
Loading…
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
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…
Previous Next
ProTip!
Follow long discussions with comments:>50.