Can AI help us predict extreme weather?