Skip to content

Actions: AI-Hypercomputer/maxdiffusion

Actions

Add Testgrid Link to PR

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
XLML.yml
212 workflow runs
212 workflow runs

Filter by Event

⌥ + click/return to exclude

Filter by Status

⌥ + click/return to exclude

Filter by Branch

⌥ + click/return to exclude

Filter by Actor

⌥ + click/return to exclude
Perf(LTX2): Comprehensive XLA, Memory, and Transformer Code Quality Optimizations Add Testgrid Link to PR #219: Pull request #422 opened by Perseus14
7s ltx2-improvements
ltx2-improvements
7s
fix(ltx2): Fix VAE timing regression for large batch sizes Add Testgrid Link to PR #218: Pull request #421 opened by mbohlool
11s fix-vae-timing-regression
fix-vae-timing-regression
11s
Fix custom flash block sizes fallback in Ulysses attention Add Testgrid Link to PR #217: Pull request #420 opened by ninatu
14s ninatu/fix_heads_per_tile
ninatu/fix_heads_per_tile
14s
Refactor WAN loading logic: granular flags & boilerplate cleanup Add Testgrid Link to PR #216: Pull request #419 opened by ninatu
9s ninatu/wan_load_params
ninatu/wan_load_params
9s
Add Stable Diffusion 1.5 support with data-parallel inference Add Testgrid Link to PR #215: Pull request #418 opened by csgoogle
12s add-sd15-support
add-sd15-support
12s
refactor Flux transformer to use scanned blocks, dynamic checkpointing, and decoupled projections Add Testgrid Link to PR #214: Pull request #417 opened by prishajain1
8s prisha/flux_training
prisha/flux_training
8s
Fix: Plumb vmem limit and bkv_compute_in to the custom kernel Add Testgrid Link to PR #213: Pull request #416 opened by eltsai
8s plumb_vmem_limit
plumb_vmem_limit
8s
feat(ltx2): make run_text_encoder_on_tpu default False and dynamically load torchax Add Testgrid Link to PR #212: Pull request #415 opened by Perseus14
6s ltx2_tpu_encoder_disable
ltx2_tpu_encoder_disable
6s
feat(ltx2): implement centralized, configuration-driven logical sharding strategy for LTX-2 and LTX-2.3 Add Testgrid Link to PR #211: Pull request #414 opened by Perseus14
6s ltx2_sharding
ltx2_sharding
6s
add g++ compiler to base image and implement hardware-aware FeedForward sharding Add Testgrid Link to PR #210: Pull request #413 opened by prishajain1
10s prisha/img_fix
prisha/img_fix
10s
Optimize Wan VAE: Add Temporal Chunking, On-TPU Postprocessing, Static JIT Support, and Configurable Dtypes Add Testgrid Link to PR #209: Pull request #411 opened by Perseus14
9s wan_vae_opt
wan_vae_opt
9s
Fix NaN in LTX2 scheduler when dynamic shift saturates sigmas before shift_terminal Add Testgrid Link to PR #208: Pull request #410 opened by flyingblackshark
6s flyingblackshark:main
flyingblackshark:main
6s
Fix XProf GCS upload bug in max_utils.Profiler Add Testgrid Link to PR #207: Pull request #409 opened by eltsai
6s xprof_upload_fix
xprof_upload_fix
6s
Add text_encoder_dtype and compile_text_encoder config parameters for Wan text encoders. Add Testgrid Link to PR #206: Pull request #408 opened by ninatu
10s ninatu/fix_text_encoder
ninatu/fix_text_encoder
10s
Fix VAE spatial sharding dynamic calculation bug in Wan pipeline. Add Testgrid Link to PR #205: Pull request #407 opened by ninatu
9s ninatu/wan_vae_sharding_bug
ninatu/wan_vae_sharding_bug
9s
feat: add pretrained orbax cache for fast WAN inference loads Add Testgrid Link to PR #204: Pull request #406 opened by csgoogle
8s wan-orbax-checkpoint-cache
wan-orbax-checkpoint-cache
8s
Fixes import errors for huggingface_hub.utils Add Testgrid Link to PR #203: Pull request #405 opened by martinarroyo
8s fix-huggingface-import
fix-huggingface-import
8s
Cancel in-progress workflow runs on new push Add Testgrid Link to PR #202: Pull request #403 opened by csgoogle
13s sagarchapara/cancel-workflows-on-push
sagarchapara/cancel-workflows-on-push
13s
Use jnp.tile instead of deprecated pltpu.repeat Add Testgrid Link to PR #201: Pull request #401 opened by eltsai
7s jax_update
jax_update
7s
feat: add KV caching support for Wan models Add Testgrid Link to PR #200: Pull request #400 opened by Perseus14
8s wan_kv_cache
wan_kv_cache
8s
Add unit test to initialize and run Wan-VACE pipeline Add Testgrid Link to PR #199: Pull request #399 opened by ninatu
10s ninatu/wan-vace-tests
ninatu/wan-vace-tests
10s
[LTX-2] Run Gemma-3 Text Encoder natively in JAX via TorchAX Add Testgrid Link to PR #198: Pull request #398 opened by mbohlool
5s text_encoder_tpu3
text_encoder_tpu3
5s
feat(wan): Add text encoder batching and optional scan loop for diffusion Add Testgrid Link to PR #197: Pull request #397 opened by Perseus14
5s wan_autoencoder_opt
wan_autoencoder_opt
5s
fix: enable ulysses sharding for custom kernels and improve scaling precision Add Testgrid Link to PR #196: Pull request #396 opened by Perseus14
6s fix-ulysses-custom
fix-ulysses-custom
6s
fix: Resolve Gemini PR review trust workspace issue and update settings Add Testgrid Link to PR #195: Pull request #395 opened by Perseus14
11s fix-gemini-review
fix-gemini-review
11s