⚡ AMD и NVIDIA: прогноз погоды на GPU
AMD показала запуск модели StormCast на GPU Instinct.
StormCast — нейросеть от NVIDIA для высокоточного прогноза гроз и конвекции с разрешением 3 км. Модель сочетает детерминированную U-Net архитектуру и генеративный diffusion-компонент для воспроизведения неопределённости погоды. Она использует данные HRRR и GFS, работает на ROCm через PyTorch в Docker-контейнере. Запуск на MI300X занимает <2 мин и 9.6 ГБ видеопамяти.
Доступны скрипты для запуска и визуализации, включая сравнение с данными NOAA.
