3 projects
crn-jax
Chemical reaction networks in JAX: GPU-parallel stochastic simulations.
myriad-jax
JAX-native platform for massively parallel control, system identification, and active learning of uncertain, stochastic systems.
gym-anm
A framework to build Reinforcement Learning environments for Active Network Management tasks in electricity networks.