https://platoaistream.net/plato-data/scale-llms-with-pytorch-2-0-fsdp-on-amazon-eks-part-2-amazon-web-services/