Machine Learning Engineer (Reinforcement Learning for UAV)

Miltech Company

Що робити

Побудова з нуля архітектури навчання від симулятора до деплою на реальне залізо.
Розробка моделей контролу на базі RL, що приймають рішення на основі відеопотоку та даних з сенсорів дрону.
Реалізація sensor fusion для забезпечення стабільного керування навіть при різких маневрах та вібраціях у взаємодії з Sensor Engineer.
Планування та участь у створенні симуляційного середовища (Isaac Gym / AirSim / UE custom на твій вибір) з глибоким Domain Randomization для роботи в складних погодних умовах (дощ, сніг, низька видимість).
Квантування та прунінг моделей для досягнення стабільного FPS на обмежених ресурсах онборд комп’ютера.

Маєш досвід з моделями PPO, SAC або TD3 та розумієш, як змусити їх працювати в real-time системах.
Розумієш, як об'єднати дані з камери та IMU (через Kalman Filters або безпосередньо в RL архітектуру) для компенсації дрейфу та затримок.
Знаєш, як наблизити фізику симулятора до реального світу, щоб мінімізувати Gap при перенесенні на борт.
Розумієш обмеження компактних онборд комп’ютерів сімейства Raspberry Pi та Nvidia та вмієш боротися за кожну мілісекунду інференсу.
Готовий самостійно вибрати стек, симулятор та підхід до навчання.

Досвід з PyTorch Mobile, TensorRT, SNPE, AIMET, Tensorflow Lite.
Досвід з БПЛА різних типів.
Досвід у UE, gamedev.
Чому варто приєднатися:
Мультидисциплінарна команда: Ти працюватимеш пліч-о-пліч із професіоналами в галузях дронобудування, GNC, Python та C++ розробниками. Ми відкриті до сміливих інновацій та нестандартних підходів.