ipython[all] ==8.25.0
numpy >=1.23.0,<2  # not yet ready for numpy 2

cuda-python

litgpt @ git+https://github.com/Lightning-AI/lit-gpt@940ffc96f7214bca24aa77479bc7c33900aaef28
