Stochastic neural simulation of plasma trajectories in nuclear fusion

This Github repository is split into various branches, each corresponding to one of the experiments/evaluations. Run the sandbox.ipynb to train and evaluate the given model.