Created by: ruanslv
Attempt to reduce memory usage from alibi embeddings. Running some ablations to test