Senior ML Engineer (AI Research, Physical AI)
Nebius
4 weeks ago
Not specifiedroboticsreinforcement learningmultimodal learning
4 weeks ago
Workplace
Not specified
Posted
4 weeks ago
Summary
Senior ml engineer role in Nebius's AI R&D team, focused on physical AI, robotic agents, multimodal foundation models, reinforcement learning, and simulation. The role involves research prototyping, large-scale training with Python and JAX, and translating approaches into reliable real-world robotic systems.
Responsibilities
- Design and train large models for robotic agents
- Develop vision-language-action architectures for physical control
- Build embodied learning datasets and quality pipelines
- Develop simulation environments and run sim-to-real experiments
- Implement planning and action-trajectory search methods
- Write research software and distributed training infrastructure
- Translate research prototypes into real-world robotic systems
pythonjaxreinforcement learningimitation learningmultimodal generative modelingcomputer visionroboticsplanningcontroltransformer-based modelsmultimodal foundation modelsvision-language-action modelsoffline rlactor-critic methodspporeward modelingpreference learningmodel-based rlmujocoisaac simisaac labpybulletrosfsdpzeroflashattentionmixed-precision trainingquantizationdistributed checkpointing
roboticsreinforcement learningmultimodal learningsim-to-real transfermodel-based controlembodied ai
