Update BC implementation, and various stuffs

[ludc]

• Add train and eval mode in RemoteAgent and NRemoteAgent
• Update BC code
• Update BRAX PPO Implementation such that it looks like other RL implementations