File size: 442 Bytes
eadc767 | 1 2 3 4 5 6 7 8 9 | V-JEPA2 encoder NKI kernels for AWS Trainium2
Copyright 2026 (kernels: res_layernorm.py, attention_block.py, mlp.py, rope_tables.py,
model glue and tooling -- original work, Apache-2.0)
kernels/attention_inner.py is a modified copy of the AWS Neuron SDK's nkilib
core/attention/attention_cte.py, Copyright Amazon.com, Inc. or its affiliates,
licensed under the Apache License, Version 2.0. See the file header for the
list of modifications.
|