fix: align step_reward with grade_episode, pin deps, update docs, clean inference

Initial Commit