fix(research): remove wandb dependency and guard optional imports

Handle checkpoints

Initial commit - add source files from research repos