Add multigpu support & UI to test embeddings

[srmsoumya]

• Update MEAN & STD for 10% of the data used for CLAY v0 model training
• Add streamlit UI with support for vector search & arithmetic
• Fix issue with tensors on different cuda devices when doing multi-gpu training