lhnguyen102 / cuTAGI

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

Temporal Dependencies for LSTM #79

Closed lhnguyen102 closed 1 month ago

lhnguyen102 commented 1 month ago

Description

This PR introduces the temporal dependencies for LSTM layer where we include the previous cell and states to compute LSTM's output. This feature is added to both CPU and CUDA version

Changes Made

Notes for Reviewer

lhnguyen102 commented 1 month ago

@van-dai-vuong If you have any comments/questions on any changes, don't hesitate to ask through the GitHub review system. I plan to close the PR today so we can release the fix

van-dai-vuong commented 1 month ago

I only have one comment, other than that, I tested. It runs good. Thanks @lhnguyen102.