Stochastic Transformer

This repository contains the code for the paper: Extending Deep Learning Emulation Across Parameter Regimes to Assess Stochastically Driven Spontaneous Transition Events

This is an evolving and experimental repositry and is subject to change.

Stochastic Transformer Model Architecture

Latidude-time plots displaying jets generated via numerical integration and by the Stochastic Transformer for different paramter values

Citations

If you use Stochastic Transformer please cite:

@inproceedings{
shokar2024extending,
title={Extending Deep Learning Emulation Across Parameter Regimes to Assess Stochastically Driven Spontaneous Transition Events},
author={Ira J. S. Shokar and Peter H. Haynes and Rich R. Kerswell},
booktitle={ICLR 2024 Workshop on AI4DifferentialEquations In Science},
year={2024},
url={https://openreview.net/forum?id=7a5gUX4e5q}
}