Re-structure caching directories for the dataset

[vratins]

split the cache into geometry, slae embeddings, esm embeddings -- instead of caching them all as a single .pt file for each pdb

[vratins]

Closed #43