Is there a Docker file for this project?

[DaDudeIan]

Hello author!

Great work on this; really interesting to read about!

I am having severe issues trying to install and run this model on my own machine. These issues largely arise from version dependencies, particularly from jaxlib. Thought I found a work around, but it did not work.

So, do you have a docker file that contains a working implementation that can be uploaded to this repo so I can try and run it myself.

Looking forward to your response.

Thank you!