Machine Learning Engineer

Job Description

Ahrefs is seeking a Machine Learning Engineer to contribute to the creation of cutting-edge AI products. The ideal candidate will be involved in turning research into real-world impact, working on projects at the forefront of AI technology.

Role involves:

• Model Development: Work on LLMs and other AI models, spanning training, fine-tuning, inference optimization, and new architectures.
• Algorithm Research & Development: Design and implement algorithms for search, agents, knowledge retrieval, and generative AI applications.
• Stay on the Cutting Edge: Stay ahead of NLP and RL advancements, evaluate emerging models/frameworks (e.g., SSMs, text diffusion), and assess their impact on our products.
• Scaling & Optimization: Optimize model inference, reduce latency, and improve efficiency using techniques like distillation, quantization, and tensor parallelism.
• Collaborate with Engineering & Product Teams: Work closely with engineering and product teams to ship AI-driven features at scale.

Requirements:

• Experience with modern deep learning frameworks (e.g., PyTorch, JAX).
• Familiarity with transformer-based architectures and techniques (e.g. flash attention, speculative decoding, KV caching).
• Strong software engineering skills (Python, CUDA, or C++ a plus).
• Interest in one or more areas: large-scale training, agentic workflows, inference optimization, retrieval-augmented generation (RAG), reinforcement learning (RLHF/RL), or multimodal AI.
• Experience with distributed training and model deployment libraries (e.g. Megatron, Ray, Triton (both or either!), vLLM) is a bonus.

Ahrefs offers:

• Opportunities to work on projects at the cutting edge of AI.
• Chance to shape the future of the product and make a real impact.