%%% Preamble Requirements %%%
% \usepackage{geometry}
% \usepackage{amsfonts}
% \usepackage{amsmath}
% \usepackage{amssymb}
% \usepackage{tikz}

% Optional packages such as sfmath set through python interface
% \usepackage{sfmath}

% \usetikzlibrary{arrows,chains,positioning,scopes,shapes.geometric,shapes.misc,shadows}

%%% End Preamble Requirements %%%

\input{"/home/jav/Python_Virt/virt_Py3.9/lib/python3.9/site-packages/pyxdsm/diagram_styles"}
\begin{tikzpicture}

\matrix[MatrixSetup]{
%Row 0
\node [DataIO] (output_data) {$\begin{array}{c}\text{data settings}\end{array}$};&
&
&
\\
%Row 1
\node [Function] (data) {$\begin{array}{c}\text{Data Generation}\end{array}$};&
\node [DataInter] (data-ode) {$\begin{array}{c}\text{model information,} \\ $$\vec{\beta}$$\end{array}$};&
&
\\
%Row 2
&
\node [Function,stack] (ode) {$\begin{array}{c}$$ODE(\vec{\beta})$$\end{array}$};&
&
\node [DataIO] (right_output_ode) {$\begin{array}{c}\text{save: trajectories,} \\ \text{plots}\end{array}$};\\
%Row 3
&
&
&
\\
};

% XDSM process chains
{ [start chain=process]
 \begin{pgfonlayer}{process} 
\chainin (data);
\chainin (ode) [join=by ProcessHVA];
\end{pgfonlayer}
}

\begin{pgfonlayer}{data}
\path
% Horizontal edges
(data) edge [DataLine] (data-ode)
(ode) edge [DataLine] (right_output_ode)
% Vertical edges
(data-ode) edge [DataLine] (ode)
(data) edge [DataLine] (output_data);
\end{pgfonlayer}

\end{tikzpicture}