Skip to content

GPU Acceleration Implementation #7

@supermlorg

Description

@supermlorg

Following is for discovery for now:

1.Matrix multiplications (Q·Kᵀ, attention scores, FFN layers)
2. Softmax
3. LayerNorm
4. Embedding + Positional Encoding
5. Batch operations

Metadata

Metadata

Assignees

Labels

No labels
No labels

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions