search

claCase / Attention-as-RNN

Non-official implementation of "Attention as an RNN" from https://arxiv.org/pdf/2405.13956, efficient associative parallel prefix scan and recurrent version implemented.
MIT License
20 stars 0 forks source link

What is the Python runtime environment, especially for the version of TensorFlow? #1

Open yanggg1133 opened 5 months ago

yanggg1133 commented 5 months ago

What is the Python runtime environment, especially for the version of TensorFlow?

claCase commented 5 months ago

Hi, I have python3.11 installed on WSL. The TensorFlow version is 2.15 with cuda, as detailed in the requirements.txt .

yanggg1133 commented 5 months ago

thinks ,good luck