ZO training support - #2

Draft

JanCSEM wants to merge 28 commits into

pulp-platform:develfrom

Commits on Feb 9, 2026

Restore 3rd branch of sleepvit
JanCSEM
committed
3c5eb6d
Swap H & W in branch2
JanCSEM
committed
477dad6
Sleep model fixes
JanCSEM
committed
10508cf
Added ZO training as Onnx4Deeploy export mode
JanCSEM
committed
e64af9e
Added Perturbation operators
JanCSEM
committed
5799139

Commits on Feb 19, 2026

Add support for exceptions, where nodes are excluded from perturbation generation
JanCSEM
committed
587e17f

Commits on Feb 20, 2026

FIX bias initialization
JanCSEM
committed
b0c10df

Commits on Feb 21, 2026

refine operators
JanCSEM
committed
eac93b3
Added Eggroll and Rademacher to ZO transform for model export
JanCSEM
committed
05da0d2

Commits on Feb 26, 2026

Split Eggroll noise generator into 2 nodes for A and B
JanCSEM
committed
4137b44
Fix shape initialiers for Eggroll
JanCSEM
committed
f4e974f

Commits on Mar 6, 2026

Modified microbenchmark tests
JanCSEM
committed
2c4e1b4

Commits on Mar 12, 2026

adding quantization support
JanCSEM
committed
5f4d683
Added weight update graph and cross entropy
JanCSEM
committed
0bccfea
update Deepquant
JanCSEM
committed
681a3d6

Commits on Mar 15, 2026

Commits on Mar 17, 2026

Add support for QSleepViT
JanCSEM
committed
1aba714

Commits on Mar 27, 2026

Update Quantization export
JanCSEM
committed
7b55f57
Adding tests to quantization passes
JanCSEM
committed
16ae86d

Commits on Mar 31, 2026

Increasing test coverage
JanCSEM
committed
029cd78

Commits on Apr 17, 2026

Add tests
JanCSEM
committed
a13abe7
Update Base exporter to use python implementation instead of ORT inference session for graphs with custom nodes
JanCSEM
committed
b14e39c
Add operator tests
JanCSEM
committed
e50793c

Commits on Jun 3, 2026

Add MCUNet, QMCUNet, TSDR, QTSDR models
JanCSEM
committed
7edea8f
Optimize and fix TSDR and QTSDR models
JanCSEM
committed
7f6492c
Update submodule
JanCSEM
committed
e24a7b3

Commits on Jun 8, 2026

update quatized models
JanCSEM
committed
b4c10b0