Hugging Face
Models
Datasets
Spaces
Buckets
new
Docs
Enterprise
Pricing
Website
Tasks
HuggingChat
Collections
Languages
Organizations
Community
Blog
Posts
Daily Papers
Hardware
Learn
Discord
Forum
GitHub
Solutions
Team & Enterprise
Hugging Face PRO
Enterprise Support
Inference Providers
Inference Endpoints
Storage Buckets
Log In
Sign Up
Kernels:
aws-neuron
/
vjepa2-neuron-kernels
Like
2
Follow
AWS Inferentia and Trainium
190
Kernel card
Files
Files and versions
xet
Community
2
main
vjepa2-neuron-kernels
/
build
/
torch-neuron
/
nki_kernels
117 kB
Ctrl+K
Ctrl+K
2 contributors
History:
1 commit
jburtoft
v1: V-JEPA2 ViT-L encoder NKI kernels for Trainium2 (4.35x, 44% MFU)
eadc767
16 days ago
__init__.py
Safe
536 Bytes
v1: V-JEPA2 ViT-L encoder NKI kernels for Trainium2 (4.35x, 44% MFU)
16 days ago
attention_block.py
Safe
41.3 kB
v1: V-JEPA2 ViT-L encoder NKI kernels for Trainium2 (4.35x, 44% MFU)
16 days ago
attention_inner.py
Safe
52.6 kB
v1: V-JEPA2 ViT-L encoder NKI kernels for Trainium2 (4.35x, 44% MFU)
16 days ago
mlp.py
Safe
12 kB
v1: V-JEPA2 ViT-L encoder NKI kernels for Trainium2 (4.35x, 44% MFU)
16 days ago
res_layernorm.py
Safe
8.91 kB
v1: V-JEPA2 ViT-L encoder NKI kernels for Trainium2 (4.35x, 44% MFU)
16 days ago
rope_tables.py
Safe
1.88 kB
v1: V-JEPA2 ViT-L encoder NKI kernels for Trainium2 (4.35x, 44% MFU)
16 days ago
