Neural Hybrid Automata: Learning Dynamics With Multiple Modes and Stochastic Transitions