270 kB

Ctrl+K

1 contributor

v1.9: padding masks on the kernel path for NeuronXLMRobertaModel (per-image additive key bias in attention_block_l1_ln_post_kb); existing models bitwise identical to v1.8

954a874 6 days ago

nki_kernels
v1.9: padding masks on the kernel path for NeuronXLMRobertaModel (per-image additive key bias in attention_block_l1_ln_post_kb); existing models bitwise identical to v1.8 6 days ago
__init__.py

1.62 kB
v1.8: post-LayerNorm encoders (NeuronXLMRobertaModel; post-LN epilogue outputs + no-RoPE switch) and LNC=2 fix for odd head-group counts 15 days ago
encoder.py

14.5 kB
v1.8: post-LayerNorm encoders (NeuronXLMRobertaModel; post-LN epilogue outputs + no-RoPE switch) and LNC=2 fix for odd head-group counts 15 days ago
layers.py

17.9 kB
v1.9: padding masks on the kernel path for NeuronXLMRobertaModel (per-image additive key bias in attention_block_l1_ln_post_kb); existing models bitwise identical to v1.8 6 days ago
metadata.json

182 Bytes
v1: NKI kernels for RoPE ViT encoders (DINOv3, V-JEPA2) on Trainium2 27 days ago