Code and training scripts for FlexOlmo - View it on GitHub