Early code

Lstm isn't working for ppo lstm.
TODO:
Fix lstm
cli for inference and training.
cli to add path of model saving
cli to use a checkpoint model for inference