Commits

Branch selector

 main

User selector

hanzhi713

Datepicker

All time

Commit History

Commits on Jan 30, 2025

• Optimizer offloading through weight-only offload (#867)

  Show description for 795da33
  hanzhi713

  authored
  795da33

  Copy full SHA for 795da33

Commits on Jan 29, 2025

• SplashAttention performance tuning for v6e (#958)

  Show description for d76ef6f
  hanzhi713

  authored
  d76ef6f

  Copy full SHA for d76ef6f

• Use InputDispatcher for fuji models (#956)

  Show description for 2d002e3
  hanzhi713

  authored
  2d002e3

  Copy full SHA for 2d002e3

Commits on Jan 28, 2025

• Add v6e PCIe overload workaround flag (#955)

  hanzhi713

  authored
  59508e3

  Copy full SHA for 59508e3

Commits on Jan 27, 2025

• Add v6e special meshes (#952)

  Show description for b125f00
  hanzhi713

  authored
  b125f00

  Copy full SHA for b125f00

Commits on Jan 23, 2025

• Legacy flash remat fix (#943)

  Show description for 30284c8
  hanzhi713

  authored
  30284c8

  Copy full SHA for 30284c8

Commits on Jan 22, 2025

• Some fixes for flash remat (#942)

  hanzhi713

  authored
  8ea85bd

  Copy full SHA for 8ea85bd

Commits on Jan 16, 2025

• Cache AoT compilation result (#927)

  Show description for 1e25e4a
  hanzhi713

  authored
  1e25e4a

  Copy full SHA for 1e25e4a

Commits on Jan 15, 2025

• Support remat for FlashAttention (#925)

  Show description for 141239c
  hanzhi713

  authored
  141239c

  Copy full SHA for 141239c

Commits on Jan 14, 2025

• Fix tf iter unit test (#924)

  hanzhi713

  authored
  3405a6e

  Copy full SHA for 3405a6e

Commits on Jan 8, 2025

• Flash2 and supports cross attention and dropout (#905)

  Show description for 6559036
  hanzhi713

  authored
  6559036

  Copy full SHA for 6559036

• Allow parallel gpu tests (#910)

  Show description for ee7d60d
  hanzhi713

  authored
  ee7d60d

  Copy full SHA for ee7d60d

Commits on Jan 2, 2025

• Implements FlashDecoding with Sparsity Support (#899)

  Show description for 60ca6ce
  hanzhi713

  authored
  60ca6ce

  Copy full SHA for 60ca6ce

Commits on Dec 24, 2024

• Fix softmax scale (#903)

  hanzhi713

  authored
  e4ff72c

  Copy full SHA for e4ff72c

Commits on Dec 6, 2024

• Support orbax state builder (#866)

  Show description for f61a269
  hanzhi713

  authored
  f61a269

  Copy full SHA for f61a269

Commits on Dec 2, 2024

• Fix feedforward remat point (#864)

  hanzhi713

  authored
  2f7291c

  Copy full SHA for 2f7291c

• fix (#852)

  hanzhi713

  authored
  c20387c

  Copy full SHA for c20387c

Commits on Nov 14, 2024

• Implement custom max_data_shard_degree and shard_threshold_bytes (#838)

  hanzhi713

  authored
  c4bbaa1

  Copy full SHA for c4bbaa1

Commits on Nov 4, 2024

• add health check (#803)

  hanzhi713

  authored
  e3293a5

  Copy full SHA for e3293a5

Commits on Nov 1, 2024

• finish todo after jax upgrade (#800)

  hanzhi713

  authored
  3b2c28d

  Copy full SHA for 3b2c28d

Commits on Oct 28, 2024

• cleanup todo (#788)

  hanzhi713

  authored
  51ab4f0

  Copy full SHA for 51ab4f0

Commits on Oct 18, 2024

• add host mount (#755)

  Show description for eb6fa3c
  hanzhi713

  authored
  eb6fa3c

  Copy full SHA for eb6fa3c

Commits on Oct 8, 2024

• fix trailing slash (#741)

  hanzhi713

  authored
  84cfc87

  Copy full SHA for 84cfc87

Commits on Oct 3, 2024

• Configurable concurrent restore gb (#720)

  Show description for f004153
  hanzhi713

  authored
  f004153

  Copy full SHA for f004153

Commits on Sep 17, 2024

• Fix race condition while doing async tf save. (#704)

  Show description for c45a018
  hanzhi713

  authored
  c45a018

  Copy full SHA for c45a018

Commits on Aug 22, 2024

• Data-sharded and zero-copy async checkpoints (#671)

  hanzhi713

  authored
  c4217ed

  Copy full SHA for c4217ed

• Asynchronous save_tf_savables (#670)

  hanzhi713

  authored
  952569f

  Copy full SHA for 952569f