Research Scientist (Embodied AI & World Models)

Role focuses on generating AI/ML ideas, designing and running experiments, and evaluating results. You will prepare, submit and present work at AI conferences and workshops. You will collaborate with researchers, silicon and software engineers to help define and test the next generation of AI hardware. Essential qualifications include Master’s/PhD or equivalent in a technical field, experience in embodied AI, world models or robotics, strong Python skills with PyTorch or JAX, solid ML fundamentals, and the ability to design, execute and report ML experiments. You should have solid mathematics (calculus, probability, linear algebra) and a track record of submitting papers to international conferences. Desirable skills include lower-level programming (C++/CUDA/Triton), hardware-aware DL concepts (threads, caches, vector/matrix engines, data dependencies), and familiarity with DL toolchains (compilation, kernel fusion, XLA/ATen, streams, asynchronous execution).