Kernels:

aws-neuron
/

vjepa2-neuron-kernels

Kernel card Files Files and versions

vjepa2-neuron-kernels / build /torch-neuron /layers.py

Commit History

Sep 30 16 days ago

19:24
v1: V-JEPA2 ViT-L encoder NKI kernels for Trainium2 (4.35x, 44% MFU)

jburtoft

eadc767