Flexible inference for animal learning rules using neural networks - Liu, Yuhan Helena; Geadah, Victor; Pillow, Jonathan (Advances in Neural Information Processing Systems, 2025)