Senior Machine Learning Engineer

Essential functions

• Research how to best use techniques like Supervised Fine Tuning (SFT) and Reinforcement Learning with Human Feedback (RLHF) to align LLMs.
• Work with technical and non-technical stakeholders to align LLMs for specific use cases.
• Research and implement approaches to ensure LLMs are aligned with product feature goals, including preventing hallucinations.
• Collect requirements from data engineering, product feature, research and evaluation teams to develop modeling, prompting and data transformation techniques

Qualifications

Required:

• Experience in processing and manipulating data using libraries such as Pandas and NumPy. Experience in visualizing data using libraries such as Matplotlib, Seaborn, or Plotly.
• Experience with deep learning frameworks - at least one of them - such as TensorFlow, PyTorch, or JAX.
• Experience in designing and experimenting with new machine learning algorithms and optimizations.
• Experience with LLM fine-tuning techniques such as Supervised Fine-Tuning (SFT), Reinforcement Learning from Human Feedback (RLHF), and parameter-efficient methods like LoRA.
• Experience in independently driving a research agenda, identifying key challenges, and proposing innovative solutions in machine learning.
• Deep understanding of Large Language Models, including their architectures, training methodologies, and inference optimizations.

Preferred:

• Experience in optimizing machine learning models and data pipelines for scalability and performance.
• Experience with software engineering principles, including version control (Git), CI/CD, and modular code design.
• Experience with distributed computing frameworks like Ray, Dask, or Spark for large-scale ML workloads.
• Experience with model deployment frameworks such as TensorFlow Serving, TorchServe, or ONNX Runtime.

We offer

• Opportunity to work on bleeding-edge projects
• Work with a highly motivated and dedicated team
• Competitive salary
• Flexible schedule
• Benefits package - medical insurance, sports
• Corporate social events
• Professional development opportunities
• Well-equipped office

About us

Grid Dynamics (NASDAQ: GDYN) is a leading provider of technology consulting, platform and product engineering, AI, and advanced analytics services. Fusing technical vision with business acumen, we solve the most pressing technical challenges and enable positive business outcomes for enterprise companies undergoing business transformation. A key differentiator for Grid Dynamics is our 8 years of experience and leadership in enterprise AI, supported by profound expertise and ongoing investment in data, analytics, cloud & DevOps, application modernization and customer experience. Founded in 2006, Grid Dynamics is headquartered in Silicon Valley with offices across the Americas, Europe, and India.