All pull requests

Pull requests

• Sol Attention Implementation

  #518·jitendra-jalwaniya opened Oct 23, 2026Collaborator·

  ·
  1

• Cast the Wan VAE weights to vae_weights_dtype instead of weights_dtype

  #517·boshjerns opened Oct 22, 2026·

  ·
  2

• Read rope_theta from rope_parameters when loading the Z-Image text encoder

  #516·boshjerns opened Oct 22, 2026·

  ·
  2

• chore: pyink-reformat two files failing the Linter workflow on main

  #515·Perseus14 opened Oct 21, 2026Collaborator·

  ·
  1

• Fix the Wan VAE attention block, which never attends over spatial positions

  #514·boshjerns opened Oct 18, 2026·

  ·
  1

• Keep the full LoRA delta when merging into bfloat16 weights

  #513·boshjerns opened Oct 18, 2026·

  ·
  6

• perf(attention): shard-major Ulysses output all-to-all (wan_ulysses_out_a2a="chunked"/"shard_major")

  pull ready
  #511·Perseus14 opened Oct 15, 2026Collaborator·9/10·

  ·
  21

• feat(wan): token-sharded patch embedding (wan_patch_embed_mode="tokens")

  #510·Perseus14 opened Oct 15, 2026Collaborator·8/10·

  ·
  7

• feat(wan): short-KV text cross-attention Pallas kernel (wan_cross_attn_kernel="pallas")

  #509·Perseus14 opened Oct 15, 2026Collaborator·7/10·

  ·
  17

• perf(wan): XLA-fused RMSNorm + RoPE producer for self-attention Q/K

  pull ready
  #508·Perseus14 opened Oct 15, 2026Collaborator·4/10·

  ·
  13

• fix(attention): dot-product fallback layout for 4-D inputs; tile_size_grid_search iters/bkv_compute guards

  #507·Perseus14 opened Oct 15, 2026Collaborator·2/10·

  ·
  6

• chore: migrate gsutil usage to gcloud storage

  #506·gurusai-voleti opened Oct 15, 2026Contributor·

  ·

• style: fix pyink formatting on main (2 files)

  #503·Perseus14 opened Oct 13, 2026Collaborator·

  ·
  1

  

• docs: add Cluster Toolkit (gcluster) guide and mark XPK as deprecated

  #501·Perseus14 opened Oct 3, 2026Collaborator·

  ·
  6

• chore: repo hygiene, CI consistency and documentation fixes

  #500·Perseus14 opened Oct 3, 2026Collaborator·

  ·
  6

• perf(wan): pad sequence to 128-lane multiple for self-attention (wan_seq_pad="lane")

  #491·Perseus14 opened Sep 25, 2026Collaborator·10/10·

  ·
  18

• SVG implementation for LTX 2

  #489·jitendra-jalwaniya opened Sep 24, 2026Collaborator·

  ·
  14

• feat(wan): fused RMSNorm+RoPE Pallas producer; Wan switches in config, not env

  pull ready
  #488·Perseus14 opened Sep 22, 2026Collaborator·6/10·

  ·
  25

• Run FLUX text encoders under Torchax instead of Flax

  #483·olehtika opened Sep 16, 2026·

  ·
  6

• Fix three FLUX regressions from the Flux.2-Klein onboarding

  #482·olehtika opened Sep 15, 2026·

  ·
  6

• feat(wan): fast serving with persistent AOT caching and tuned inference recipe

  pull ready
  #479·Perseus14 opened Sep 13, 2026Collaborator·5/10·

  ·
  36

  

• feat(attention): 2D Ulysses+Ring custom attention with fixed-m (pre-a2a norm reduction, unpadded ring K/V)

  pull ready
  #478·Perseus14 opened Sep 13, 2026Collaborator·3/10·

  ·
  27

  

• feat(attention): fixed-m splash attention kernel with dynamic bounds and safety fallbacks

  pull ready
  #477·Perseus14 opened Sep 13, 2026Collaborator·1/10·

  ·
  52

  

• fix(deps): skip nvidia-cuda-cccl on macOS (no wheels available)

  pull ready
  #476·harkanwalbedi opened Sep 11, 2026·

  ·
  1

  3

• fix: prevent silent double-application of LoRA weights

  #473·WangXukang-cypher opened Sep 4, 2026·2 of 3 tasks·

  ·
  12