tensorflow-alpa
tensorflow-alpa copied to clipboard
alpa-projects
To avoid the divergence of our codebase and tensorflow/jax mainline, we want to rebase our repos to the tensorflow/jax mainline regularly. This brings several advantages: 1. Get the latest updates...
[WIP] support tuple-shaped parameters for while instruction
# Incompatible changes - `tensorflow/compiler/xla/python/xla_compiler.cc`. The return value of `spmd_output_sharding` and `spmd_parameters_shardings` is not converted to tuple. - `tensorflow/compiler/xla/service/hlo_verifier.cc`. Temporarily disable the verification on fused instruction due to our CommonComputationElimination...