Refactor Hackable Diffusion for improved XLA performance with Flash Attention and fused RMS normalization.#67
Open
copybara-service[bot] wants to merge 1 commit into
Open
Refactor Hackable Diffusion for improved XLA performance with Flash Attention and fused RMS normalization.#67copybara-service[bot] wants to merge 1 commit into
copybara-service[bot] wants to merge 1 commit into
