activation / build

Commit History

feat: support reset_parameters()
605f22e

iamwyldecat commited on

feat(rms-norm): Impl fused RMSNorm
f3b99fb

iamwyldecat commited on

feat(poly-norm): add perf test
d14fd4d

iamwyldecat commited on

fix(poly-norm): calc param grad explicitly
704692b

iamwyldecat commited on

fix(poly-norm): fix bug in reduce sum
883cc1c

iamwyldecat commited on

refactor(poly-norm): use const for immutable args
e85ecc9

iamwyldecat commited on

chore: use latest build image and misc
f5a7d38

iamwyldecat commited on

chore(poly-norm): update README and build artifacts
f72121c

iamwyldecat commited on

chore(poly-norm): add ROCm build artifacts
4b70498

iamwyldecat commited on