Thought you might be interested – it is definitely a platform issue with Linux on AWS. (I only tried Ubuntu, don’t know about other flavors of Linux).
See other reports, e.g.
On AWS Linux, running the notebook linked earlier in the thread allocates >> 244 GB and kills the machine.
On AWS Windows, the process is stable at 160MB. Not 160GB, but 160MB.
Thanks for your attention to this. I will file an issue on Github if you want me to, please let me know.