litgpt[all]==0.5.3 fire scikit-learn matplotlib jupyter nbclassic psrcal expected_cost