"""CausalCellJEPA data and model components."""