[TPU][Pallas] relax tolerances and fix Pallas autotuning OOM in layer_norm#2272
Draft
yarongmu-google wants to merge 1 commit into
Draft
[TPU][Pallas] relax tolerances and fix Pallas autotuning OOM in layer_norm#2272yarongmu-google wants to merge 1 commit into
yarongmu-google wants to merge 1 commit into