lightning>=2.0.0
jsonargparse[all]
scikit-learn
speed-benchmark
