First, thank you very much for the nice JAX implementation of Hyena as well as the recursive Hyena-S5!
As I'm trying to reproduce the results, there are several environments issues.
Just wonder if it's convenient to ask for the CUDA version as well as the python packages configurations for reproducing the results.
First, thank you very much for the nice JAX implementation of Hyena as well as the recursive Hyena-S5!
As I'm trying to reproduce the results, there are several environments issues. Just wonder if it's convenient to ask for the CUDA version as well as the python packages configurations for reproducing the results.
Thanks!