Machine Learning Engineer - Python

Description :

We're seeking a Machine Learning Engineer who wants to operate at the frontier of generative search, evaluation, and optimization. You'll design and deploy models that simulate and analyze how LLMs surface information, training systems that measure, predict, and improve AI visibility. Your work will bridge research and production : building GEO evaluation frameworks, training models that make generative engines more interpretable and optimizable, and designing data-efficient LLM simulators.

Responsibilities :

- Build machine learning systems that model and predict LLM search and generation behaviors.

- Design and train models for ranking, retrieval, and generative evaluation tasks.

- Develop GEO-Bench datasets and metrics for measuring visibility across AI engines.

- Collaborate with researchers to prototype and productionize novel learning algorithms.

- Optimize inference pipelines for efficiency, cost, and latency.

Requirements :

- Strong proficiency in Python and ML frameworks (PyTorch, TensorFlow, or JAX).

- Experience with large-scale training and inference (Ray, Triton, Vertex AI, or Hugging Face).

- Familiarity with evaluation frameworks (MLflow, Weights and Biases) and data pipelines.

- Solid grounding in applied ML or NLP retrieval, ranking, or text generation.

Preferred :

- Experience with LLM fine-tuning, prompt optimization, or RAG architectures.

- Familiarity with black-box optimization or reinforcement learning.

- Background in generative model evaluation, interpretability, or search relevance.