Founding ML Engineer — RL & Scale Training

Axiōma Search is seeking a skilled ML engineer to advance post-training methods for large-scale AI systems. You will improve training loops from rollout to policy updates, enhance throughput and GPU efficiency, and diagnose bottlenecks across distributed environments.

You’ll bridge ML research with systems engineering to ensure reliable, scalable model training. You will work on reinforcement learning and related post-training strategies, using RL approaches like PPO, GRPO and SDPO, and you’ll