There's a rust deep learning library called dfdx that just setup llama: https://github.com/coreylowman/llama-dfdx