search

lhnguyen102 / cuTAGI

CUDA implementation of Tractable Approximate Gaussian Inference
MIT License
24 stars 9 forks source link

CPU version for Self Attention Layer #34

Closed lhnguyen102 closed 1 year ago

lhnguyen102 commented 1 year ago

Goal

The goal is to implement the self-attention mechanism based on TAGI approach.

Task