Models
Datasets
Spaces
Buckets new
Docs
Enterprise
Pricing
- Website
- Community
- Solutions
Log In
Sign Up

Kernels:

aws-neuron
/

vit-encoder-neuron-kernels

Kernel card Files Files and versions

vit-encoder-neuron-kernels / build /torch-neuron

251 kB

Ctrl+K

Ctrl+K

1 contributor

History: 8 commits

jburtoft's picture

CHANGELOG and comment wording

6da9fa8 about 7 hours ago

nki_kernels
CHANGELOG and comment wording about 7 hours ago
__init__.py

1.36 kB
v1.4: 5-bank QK^T PSUM rotation (<=1024 tokens) + experimental opt-in FP8 MLP 2 days ago
encoder.py

14 kB
v1.6: two images per producer pass in attention_block (LNC=2 path; LNC=1 single-group models) 1 day ago
layers.py

9.81 kB
v1: NKI kernels for RoPE ViT encoders (DINOv3, V-JEPA2) on Trainium2 7 days ago
metadata.json

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