AI Researcher — Training Optimization

Remote, Worldwide

About the Role

We’re looking for an AI Researcher focused on training optimization to help us push the efficiency, stability, and scalability of large-scale model training. You’ll work at the intersection of research and systems , developing novel techniques to reduce training cost, accelerate convergence, and improve model quality—while validating ideas through rigorous experiments and publications.

This role is ideal for someone who enjoys turning research insights into practical training wins , and who has a track record (or strong ambition) of publishing applied ML research .

What You’ll Work On

Design and evaluate training optimization techniques for large models (e.g. optimization algorithms, schedulers, normalization, curriculum strategies)

Improve training efficiency and stability across long runs and large datasets

Research and implement methods such as:

Optimizer and scheduler innovations

Mixed-precision, low-precision, and memory-efficient training

Gradient noise reduction, scaling laws, and convergence analysis

Training-time regularization and robustness techniques

Run large-scale experiments, analyze results, and translate findings into actionable...