Feat/triton trilinear#14

Open

ruhmamehek wants to merge 10 commits intoGeWu-Lab:mainfrom

ruhmamehek:feat/triton-trilinear

Commits on Feb 21, 2026

Add AVE reproducibility updates and gradient sensitivity analysis tooling
Ruhma Mehek Khan
committed

Commits on Feb 23, 2026

Fix grad sensitivity group aggregation and add math derivation doc
Ruhma Mehek Khan
committed

Commits on Mar 1, 2026

BF16 AVE training setup + latest debugging updates
Ruhma Mehek Khan
committed
BF16 AVE training setup + latest debugging updates
Ruhma Mehek Khan
committed

Commits on Mar 5, 2026

added changes in code for ablation analysis
Ruhma Mehek Khan
committed
cast weights to match dtypes
ruhmamehek
committed
added eval scripts, updated run saving scheme
ruhmamehek
committed
updated config
ruhmamehek
committed

Commits on Mar 11, 2026