ipython[all] ~=8.37.0
numpy
liger-kernel == 0.4.0
cuda-python
litgpt == 0.5.1
