Pre-trained and transformer Neural Network models in Axon

Simple LoRA implementation for fine-tuning Axon models