AI for weather predictions