aba83800 · [xla:gpu] Simplify the search for operand slices in DynamicAddressComputationFusion · 3月 20, 2024

1dd6bcfd · [PJRT C API] Plumb plugin attributes from plugin to JAX python. · 3月 20, 2024

076613b1 · [XLA:Runtime] Moved the nccl_collective_broadcast thunk to new folder and... · 3月 20, 2024

e249ab57 · Add checks for output file size ratio in the integration tests · 3月 20, 2024

2b21034a · [xla:gpu] Support DUS in DynamicAddressComputationFusion · 3月 20, 2024

05073bb5 · [PJRT][IFRT] Update PJRT, IFRT, and Py executable getters to return PjRtLayouts · 3月 20, 2024

cfa49e5a · [xla:gpu][NFC] Simplify HloPredicateIsOp · 3月 20, 2024

2a5cc7ee · Move `tsl/python` to `xla/tsl/python` · 3月 20, 2024

6dcedc11 · Add new overload of `GraphToFunctionDef()` that can consume the graph. · 3月 20, 2024

a1524048 · Migrate to new header · 3月 20, 2024

696cdf46 · To ensure that C++20 is fully supported in TensorFlow, · 3月 20, 2024

2666ad89 · Flag Reserved GPU HBM Size in Pathways. · 3月 20, 2024

efbaf109 · Add support for 'tflite_data' parameter to the 'cc_test_with_tflite' rule. · 3月 20, 2024

3ae4bed2 · Guard host transfers inside pure_callbacks from deadlocking the TPU. · 3月 20, 2024

897af8f9 · Replace `RemoteTensorHandle` with `TensorProto` for scalars in an... · 3月 20, 2024

72d72a52 · [MSA] Refactor AlternateMemoryBestFitHeap out of memory_space_assignment. · 3月 20, 2024

349b1416 · Clean up some deps in flatbuffer_import, flatbuffer_export. · 3月 20, 2024

b7946f42 · Add debug logging when resharding may be needed. · 3月 20, 2024

a0d17dbc · Allow Shardonnay to use XLA targets. · 3月 20, 2024

b2efa83c · Log checkpoint hashing error. · 3月 20, 2024