Kernels
activation / torch-ext

Commit History

add readme with precommit hooks and applied pre commit to all files
f517c97

TaehyunKimMotif commited on

feat: support reset_parameters()
605f22e

iamwyldecat commited on

feat(rms-norm): Impl fused RMSNorm
f3b99fb

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

feat(poly-norm): add default value for eps argument
afd2a56

iamwyldecat commited on

feat(poly-norm): Add PolyNorm
44e9845

iamwyldecat commited on