Closed lennart-finke closed 2 weeks ago
Alright, will have to work on this some more as the checks indicate.
Now it might go through. If someone has time, I recommend double checking the jaxtyping hints though, as I was not entirely sure everywhere.
Description
Added Jax and regular typing to
llama_train.py
Related Issue
Should close #4.
How Has This Been Tested?
Executing the script in a Colab instance.