Machine Learning Engineer – Digital Intelligence

What you'll do

• Research and prototype architectures and pre-training objectives for structured, unstructured, and heterogeneous data.
• Implement research ideas in production-quality code and optimize large-scale training throughput.
• Collaborate on product solutions and mentor engineers on ML best practices.

What they ask for

• Master’s degree with 2+ years or Bachelor’s degree with 4+ years in Computer Science, with training and work experience in Machine Learning, LLM/NLP, or similar fields.
• Deep LLM and Transformer expertise, including attention mechanisms, RoPE, and multimodal inputs.
• PyTorch proficiency at scale, including distributed training with FSDP and DeepSpeed and practical memory optimization.
• Experience pre-training foundation models from scratch and designing tokens and vocabularies for complex data; strong production software engineering skills; prior experience with financial data and recommendation systems.

Nice to have

• Publication record at top AI/ML venues.
• Experience with post-training LLMs, serving infrastructure, network optimization, interpretability or steering, large-scale compute, or shipping real-world products.

Pay

USD 137,750.00-$195,000.00 base salary

Tools

LLM, Transformer, RoPE, PyTorch, FSDP, DeepSpeed

AI focus

Research, train, optimize, and deploy foundation models and LLMs, including tailored pre-training, multimodal data handling, and model quality and performance optimization.