Add initial_weights support for fine-tuning#7

sirmarcel merged 1 commit intomainlab-cosmo/lorem-jax:mainfrom

feature/initial-weightslab-cosmo/lorem-jax:feature/initial-weightsCopy head branch name to clipboard