ESM3 dataset optimization