Staff ML Engineer, Fine Tuning - Slack

About the role

Slack is seeking a Staff Machine Learning Engineer to join its ML team. This role focuses on designing, training, and deploying NLP models that enhance core product experiences.

Responsibilities

• Design and execute finetuning strategies for large language models and other deep learning architectures tailored to Slack's NLP tasks.
• Own the model training lifecycle end-to-end, including data curation, training infrastructure, hyperparameter optimization, evaluation, deployment, and monitoring.
• Build and maintain scalable finetuning training pipelines on GPU infrastructure.
• Collaborate with Product Managers, Designers, and Frontend Engineers to develop new features for the growing user base.
• Lead large multi-functional projects that significantly impact the business.
• Mentor other engineers and deeply review code.
• Improve engineering standards, tooling, and processes.

Requirements

• 5+ years of hands-on experience training and fine-tuning deep learning models in NLP (or a closely related domain).
• Experience with common deep learning frameworks like PyTorch, TensorFlow, JAX, etc.
• An analytical and data-driven mindset, with experience measuring success with complex ML/AI products.
• Experience with functional or imperative programming languages.
• Strong communication skills and ability to explain complex technical concepts to designers, support, and other specialists.

Qualifications

• Track record of shipping fine-tuned models to production that serve real users at scale.
• Expertise with recommendation systems or search.
• Familiarity with model optimization for inference (quantization, pruning, speculative decoding, compilation via TorchScript/TensorRT/ONNX).
• Experience with retrieval-augmented generation and hybrid retrieval/generation systems.
• Broad experience across NLP, ML, and Generative AI capabilities.
• Knowledge of using multiple data types in RAG solutions including structured, unstructured, and knowledge graphs.
• Broad experience across NLP, ML, and Generative AI capabilities.

Skills

• Deep training and productionization expertise.
• Hands-on experience with model training and finetuning.
• Experience with common deep learning frameworks.
• Experience with functional or imperative programming languages.
• Experience with model optimization for inference.
• Experience with retrieval-augmented generation and hybrid retrieval/generation systems.
• Broad experience across NLP, ML, and Generative AI capabilities.

Benefits

At Salesforce, you can expect a range of benefits including:

• Time off programs
• Medical, dental, vision, and mental health support
• Paid parental leave
• Life and disability insurance
• 401(k)
• Employee stock purchasing program

For more details, visit Salesforce Benefits.

The typical base salary range for this position is $197,300 - $313,700 annually. In select cities within the San Francisco and New York City metropolitan area, the base salary range is $237,700 - $344,700 annually.