Оба варианта правильны. «В мозге» обычно используется в анатомическом смысле: опухоль в мозге. «В мозгу» синоним фразы «в голове»: у него в мозгу промелькнула мысль = у него в голове промелькнула мысль.

В русском языке есть Локатив (или местный падеж). Его основная функция - это функция места. Обычно форма Локатива совпадает с Препозитивом (Prepositional case). Но не всегда. Есть группа слов, где окончание будет другое: сад, порт, угол, лес ... и мозг. Где? В мозгУ Где? В лесУ Где? В портУ Но, если мы говорим не о функции /места/, а о функции /тема разговора/, то это уже будет Предложный падеж и знакомые нам окончания: Думаю о чём? О лесЕ Говорю о чём? О мозгЕ

Для меня звучит "в мозге" лучше.🤓