UAI-MP-AIF-JAX

Converts a standard Bethe BP implementation to the Active Inference (AIF-MP) message-passing scheme. Specifies four channel reparameterizations, modified factor kernels, and correct fixed-point equations. Reference paper Appendix D (T=1 derivation) and D.6 (generic T).

Expert in implementing discrete message-passing schemes using JAX primitives. Translates LaTeX update rules into indexed or tensor-based contractions for state-space models and planning. Covers the full method family from standard BP to AIF-MP.

Converts a standard Bethe BP implementation to the VBP (cross-entropy planning) message-passing scheme. Single action channel in the dynamics numerator, unmodified observation kernel, theta inferred via cavities. Reference paper Appendix E.

Generate Slurm job scripts and submission scripts for the Snellius HPC cluster from DVC pipeline stages. Use this skill whenever the user mentions "cluster", "snellius", "slurm", "sbatch", "submit job", "run on cluster", "HPC", or wants to run experiments on a remote GPU cluster. Also trigger when the user adds new DVC stages and might need corresponding cluster scripts, or asks about job submission, array jobs, or cluster resource allocation.