Kernels:

aws-neuron
/

vit-encoder-neuron-kernels

Kernel card Files Files and versions

vit-encoder-neuron-kernels

286 kB

Ctrl+K

Ctrl+K

1 contributor

History: 10 commits

jburtoft's picture

v1.6: two images per producer pass in attention_block (LNC=2 path; LNC=1 single-group models)

a827ebd about 14 hours ago

build
v1.6: two images per producer pass in attention_block (LNC=2 path; LNC=1 single-group models) about 14 hours ago
examples
examples: use aws-neuron/vit-encoder-neuron-kernels repo id (docs only) about 16 hours ago
.gitattributes

211 Bytes
v1: NKI kernels for RoPE ViT encoders (DINOv3, V-JEPA2) on Trainium2 6 days ago
CHANGELOG.md

11 kB
v1.6: two images per producer pass in attention_block (LNC=2 path; LNC=1 single-group models) about 14 hours ago
LICENSE

11.4 kB
v1: NKI kernels for RoPE ViT encoders (DINOv3, V-JEPA2) on Trainium2 6 days ago
NOTICE

961 Bytes
v1: NKI kernels for RoPE ViT encoders (DINOv3, V-JEPA2) on Trainium2 6 days ago
README.md

11 kB
README: point to aws-neuron/vjepa2-neuron-kernels (repo moved to aws-neuron org; docs only) about 16 hours ago