Commits

Branch selector

 main

User selector

All users

All time

Commit History

Commits on Sep 15, 2025

• fix error: conv_general_qt requires the same bwd_qtype as weight_qtype (#245)

  Show description for 3cd6a44
  susanbao

  authored
  3cd6a44

  Copy full SHA for 3cd6a44

Commits on Sep 12, 2025

• Update Python version to 3.12, Update WAN to match new Flax version (#244)

  Show description for fd53cde
  susanbao

  authored
  fd53cde

  Copy full SHA for fd53cde

Commits on Sep 11, 2025

• Wan flops calc (#243)

  Show description for c3bf323
  entrpn

  authored
  c3bf323

  Copy full SHA for c3bf323

Commits on Sep 8, 2025

• Do unboxing only on concrete states and not abstract_state (#242)

  Rohan-Bierneni

  authored
  8fdf3c2

  Copy full SHA for 8fdf3c2

Commits on Sep 5, 2025

• change gs video path to run_name/videos (#241)

  susanbao

  authored
  c24a86b

  Copy full SHA for c24a86b

Commits on Sep 3, 2025

• Add dropout (#240)

  Show description for 043f826
  entrpn

  authored
  043f826

  Copy full SHA for 043f826

Commits on Aug 28, 2025

• support generating video for eval (#239)

  Show description for 34a9134
  susanbao

  authored
  34a9134

  Copy full SHA for 34a9134

Commits on Aug 26, 2025

• Wan checkpointing (#238)

  Show description for 39b227b
  coolkp

  authored
  39b227b

  Copy full SHA for 39b227b

Commits on Aug 25, 2025

• Solve the qwix bugs when config.per_device_batch_size * jax.local_device_count() < 1 (#237)

  Show description for 224a951
  susanbao

  authored
  224a951

  Copy full SHA for 224a951

Commits on Aug 22, 2025

• adds segment ids for masking. (#236)

  Show description for 955bd86
  entrpn

  authored
  955bd86

  Copy full SHA for 955bd86

Commits on Aug 15, 2025

• Fix wan qwix test (#235)

  coolkp

  authored
  aad9839

  Copy full SHA for aad9839

• Add nightly JAX AI Image build to maxdiffusion (#234)

  Rohan-Bierneni

  authored
  d028efa

  Copy full SHA for d028efa

• B 436918994 (#233)

  Show description for b0bd1b0
  entrpn

  authored
  b0bd1b0

  Copy full SHA for b0bd1b0

• Img2vid pipeline (#227)

  Show description for c0f89e6
  Serenagu525

  authored
  c0f89e6

  Copy full SHA for c0f89e6

Commits on Aug 14, 2025

• add WAN evaluation pipeline (#232)

  Show description for 38283b5
  susanbao

  authored
  38283b5

  Copy full SHA for 38283b5

Commits on Aug 13, 2025

• Logging Fix (#222)

  Show description for c44f0e5
  Serenagu525

  authored
  c44f0e5

  Copy full SHA for c44f0e5

• fix wan unit test bugs (#231)

  Show description for ee53ee3
  susanbao

  authored
  ee53ee3

  Copy full SHA for ee53ee3

• Unit test fix (#230)

  Show description for f279995
  coolkp

  authored
  f279995

  Copy full SHA for f279995

• gather videos for multihost (#229)

  entrpn

  authored
  fa69260

  Copy full SHA for fa69260

• Rectified Flow Scheduler Test (#220)

  Show description for 6ec20ea
  Serenagu525

  authored
  6ec20ea

  Copy full SHA for 6ec20ea

Commits on Aug 12, 2025

• qwix quantize WAN transformer (#226)

  Show description for de60c6c
  susanbao

  authored
  de60c6c

  Copy full SHA for de60c6c

• Update UnitTests.yml (#228)

  Show description for ecd3514
  coolkp

  authored
  ecd3514

  Copy full SHA for ecd3514

Commits on Aug 7, 2025

• Revert "Use mediapy (#219)" (#224)

  Show description for f31d659
  coolkp

  authored
  f31d659

  Copy full SHA for f31d659

• Use mediapy (#219)

  Show description for 8ee052e
  coolkp

  authored
  8ee052e

  Copy full SHA for 8ee052e

Commits on Aug 5, 2025

• Video training part 2 (#218)

  entrpn

  authored
  fc46fcc

  Copy full SHA for fc46fcc

Commits on Aug 1, 2025

• Fix flops calculation for multi host and also gcs bucket in end2end test (#217)

  coolkp

  authored
  8274aca

  Copy full SHA for 8274aca

Commits on Jul 30, 2025

• ltxvid instruction update (#214)

  Show description for deeb20b
  Serenagu525

  authored
  deeb20b

  Copy full SHA for deeb20b

Commits on Jul 28, 2025

• LTXVid text2vid pipeline (#208)

  Show description for 80771b1
  Serenagu525

  authored
  80771b1

  Copy full SHA for 80771b1

• Updating getting started guide to point users to use a more recent ve… (#194)

  Show description for 76f84a1
  Grappeggia

  authored
  76f84a1

  Copy full SHA for 76f84a1

• use local_devices instead of devices which defaults to first machine's devices. (#212)

  entrpn

  authored
  9b07874

  Copy full SHA for 9b07874

• Make Maxdiffusion compatile w/ JAX 0.7.0 and latest orbax version (#211)

  Show description for e556ca1
  Rohan-Bierneni

  authored
  e556ca1

  Copy full SHA for e556ca1

Commits on Jul 25, 2025

• Fix flux inference and sdxl training (#210)

  coolkp

  authored
  4e0999c

  Copy full SHA for 4e0999c

Commits on Jul 24, 2025

• Video dataset (#209)

  Show description for 1e1d2e1
  entrpn

  authored
  1e1d2e1

  Copy full SHA for 1e1d2e1

Commits on Jul 18, 2025

• fixes flux training. (#206)

  entrpn

  authored
  db4caf0

  Copy full SHA for db4caf0

• Remove mlperf logging (#205)

  Show description for 12469fa
  coolkp

  authored
  12469fa

  Copy full SHA for 12469fa