Gradient-based optimization of exact stochastic kinetic models
This paper introduces a gradient-based optimization method using straight-through Gumbel-Softmax estimation to enable efficient parameter inference and inverse design in exact stochastic kinetic models by approximating gradients through continuous relaxation while preserving discrete stochastic dynamics in the forward pass.