Staff Software Engineer - Machine Learning Platform

The ML Platform team is the foundation of ML and Data Science at Discord. Our aim is to create the best experience throughout the machine learning workflow, from training to serving. We supplement our platform by building core and widely applicable datasets, models, and services, all to help our users find belonging in a safe and welcoming space.

You will build the Machine Learning platform at Discord, working with everything from feature stores, real-time data processing, LLM tooling, to model serving at scale. You will lead projects and work directly with ML practitioners. You have experience building Machine Learning Platforms, putting Machine Learning into production, and an understanding of ML models and techniques. You will report to the Engineering Manager of the ML Platform team. This is a great chance to help take machine learning into new and exciting areas to the benefit of our 150M users.

What you'll do

• Design and build the platform ML engineers and data scientists use to understand and delight Discord's users and keep them safe.

• Evaluate and integrate new machine learning frameworks and tools to ensure that Discord keeps up with the fast moving world of ML, including LLMs and generative AI

• Collaborate with model builders to ensure we have a smooth path from idea to production

• Set best practices in machine learning at Discord

• Create foundational datasets and models

What you should have

• 8+ years of experience working as a software engineer in data or backend with exposure to large datasets or distributed systems

• 4+ years working on platforms or infrastructure

• 2+ years working on machine learning platforms

• You have experience with orchestration systems (such as Airflow, Dagster, or Argo)

• You have put machine learning models into production

Bonus Points

• Experience with real-time data processing (Spark, Flink, Dataflow, Kafka, Pulsar, etc.)

• Experience debugging and maintaining live production systems on Kubernetes

• Experience building ML models in TensorFlow or PyTorch

The US base salary range for this full-time position is $235,000 to $256,000 + equity + benefits. Our salary ranges are determined by role and level. Within the range, individual pay is determined by additional factors, including job-related skills, experience, and relevant education or training. Please note that the compensation details listed in US role postings reflect the base salary only, and do not include equity, or benefits.#buildbelonging #LI-Remote #LI-Hybrid #LI-WK1