Commits

Breadcrumbs

History for

onmaster

User selector

All users

All time

Commit History

Commits on May 18, 2026

• Fix E721 type comparisons (#2003)

  LeSingh1

  authored
  33254f2

  Copy full SHA for 33254f2

Commits on Apr 20, 2026

• remove tests and examples of apex.contrib.fmha and apex.contrib.multihead_attn (#1995)

  Aidyn-A

  authored
  1d308e9

  Copy full SHA for 1d308e9

Commits on Mar 5, 2026

• Fix divide-by-zero in GroupNorm two-pass kernel for large batch sizes (#1984)

  Show description for dbe421e
  yuantailing

  authored
  dbe421e

  Copy full SHA for dbe421e

Commits on Jan 21, 2026

• Add VRAM estimation and memory-aware skipping to GroupNorm large tensor tests (#1976)

  Show description for 9c5bd72
  

  Copilot

  and

  crcrpar

  authored
  9c5bd72

  Copy full SHA for 9c5bd72

Commits on Jan 13, 2026

• Remove `apex.transformer` (#1968)

  Show description for 4fcca77
  

  syedshazli

  and

  pre-commit-ci[bot]

  authored
  4fcca77

  Copy full SHA for 4fcca77

Commits on Nov 28, 2025

• Enable Ruff pre-commit hooks (#1957)

  Show description for 16fadfe
  

  3 peopleauthored
  16fadfe

  Copy full SHA for 16fadfe

Commits on Nov 14, 2025

• fix groupnorm int32 index overflow (#1845)

  Show description for 10417ac
  

  tlogn

  and

  crcrpar

  authored
  10417ac

  Copy full SHA for 10417ac

Commits on Oct 28, 2025

• Fix contrib openfold_triton tests (#1938)

  Show description for d0e7be1
  

  crcrpar

  and

  Copilot

  authored
  d0e7be1

  Copy full SHA for d0e7be1

Commits on Sep 25, 2025

• Update group norm license. (#1935)

  alpha0422

  authored
  4fd4f30

  Copy full SHA for 4fd4f30

Commits on Jun 26, 2025

• replace `apex.parallel.flat_dist_call` in test with coalescing manager (#1913)

  Show description for d8200a1
  crcrpar

  authored
  d8200a1

  Copy full SHA for d8200a1

Commits on Jun 25, 2025

• fmha: add sm_100 and sm_120 (#1890)

  Aidyn-A

  authored
  a255aa6

  Copy full SHA for a255aa6

• Update test_fast_layer_norm.py (#1911)

  eqy

  authored
  6445b06

  Copy full SHA for 6445b06

Commits on May 15, 2025

• [Contrib][Openfold Triton] Use `json` instead of `pickle` (#1900)

  Show description for e74a67b
  

  Aidyn-A

  and

  crcrpar

  authored
  e74a67b

  Copy full SHA for e74a67b

Commits on Apr 10, 2025

• Fix GroupNorm test: properly check graph breaks (#1894)

  yuantailing

  authored
  badf311

  Copy full SHA for badf311

Commits on Mar 26, 2025

• [contrib] Add group_norm_v2 (#1887)

  Show description for c02c6c8
  yuantailing

  authored
  c02c6c8

  Copy full SHA for c02c6c8

Commits on Feb 26, 2025

• adjust threshold for `test_encdec_multihead_attn.py` (#1884)

  Fuzzkatt

  authored
  0024663

  Copy full SHA for 0024663

Commits on Nov 21, 2024

• Update test_group_norm.py (#1858)

  eqy

  authored
  97d8a6b

  Copy full SHA for 97d8a6b

Commits on Aug 30, 2024

• Add Unittest For Distributed Adam With CUDA Graph (#1836)

  Show description for b7a4acc
  

  alpha0422

  and

  crcrpar

  authored
  b7a4acc

  Copy full SHA for b7a4acc

• Traceable GroupNorm (#1835)

  Show description for 2d2db12
  

  alpha0422

  and

  crcrpar

  authored
  2d2db12

  Copy full SHA for 2d2db12

Commits on Jun 29, 2024

• deprecate uses of torch.cuda.amp (#1813)

  Show description for f8e60c4
  Fuzzkatt

  authored
  f8e60c4

  Copy full SHA for f8e60c4

Commits on Apr 24, 2024

• NCCL userbuffer for DP RS in DistOpt (#1797)

  Show description for f3f0492
  

  WanZzzzzz

  and

  Qiyu Wan

  authored
  f3f0492

  Copy full SHA for f3f0492

Commits on Feb 8, 2024

• Support scaled optimizer state in distributed Adam optimizer (#1771)

  Show description for b496d85
  timmoon10

  authored
  b496d85

  Copy full SHA for b496d85

Commits on Feb 7, 2024

• Skip the p2p test on single GPU platforms (#1775)

  nWEIdia

  authored
  7e239f7

  Copy full SHA for 7e239f7

Commits on Jan 18, 2024

• Update test_bottleneck_module.py - Skip BottleNeck Peer Memory Test if Not Supported (#1769)

  Show description for 6c8f384
  nWEIdia

  authored
  6c8f384

  Copy full SHA for 6c8f384

Commits on Jan 12, 2024

• Update test_transducer_joint.py (#1767)

  Show description for 48c4894
  nWEIdia

  authored
  48c4894

  Copy full SHA for 48c4894

Commits on Jan 10, 2024

• Increase tolerance to workaround tolerance issues on A100 (#1766)

  Show description for e9789cc
  nWEIdia

  authored
  e9789cc

  Copy full SHA for e9789cc

Commits on Dec 15, 2023

• [contrib] Improve FusedAdamSWA interface and add unit tests (#1759)

  Show description for ccffcc4
  

  lirundong

  and

  crcrpar

  authored
  ccffcc4

  Copy full SHA for ccffcc4

Commits on Nov 20, 2023

• Distributed optimizer support for contiguous param buffer with FP8 params (#1749)

  Show description for a2f6683
  timmoon10

  authored
  a2f6683

  Copy full SHA for a2f6683

Commits on Nov 8, 2023

• Port changes until 108e6739.

  Show description for 464b8f1
  

  3 peopleauthored and

  crcrpar

  committed
  464b8f1

  Copy full SHA for 464b8f1

Commits on Oct 19, 2023

• Specify `rtol` as well as `atol` in test_fmha.py (#1743)

  eqy

  authored
  9fc94b7

  Copy full SHA for 9fc94b7

Commits on Oct 17, 2023

• Use `torch.testing.assert_close` for better signals (#1742)

  eqy

  authored
  19cc873

  Copy full SHA for 19cc873

Commits on Oct 13, 2023

• add test for openfold triton mha kernel (#1739)

  Show description for 5afc6a0
  azazhu

  authored
  5afc6a0

  Copy full SHA for 5afc6a0

Commits on Sep 29, 2023

• Distributed optimizer infrastructure for FP8 parameters (#1723)

  Show description for 2386a91
  

  timmoon10

  and

  crcrpar

  authored
  2386a91

  Copy full SHA for 2386a91

Commits on Sep 28, 2023

• Massively reduce LayerNorm/RMSNorm GPU memory usage in modern networks by tricking torch autograd (#1715)

  Show description for 6ff4548
  RuiWang1998

  authored
  6ff4548

  Copy full SHA for 6ff4548

Commits on Sep 6, 2023

• Distributed optimizer support for multiple dtypes (#1721)

  Show description for 52e18c8
  timmoon10

  authored
  52e18c8

  Copy full SHA for 52e18c8