
Staff Software Engineer, Machine Learning
Discord
full-time
Posted on:
Location Type: Remote
Location: Remote • California • 🇺🇸 United States
Visit company websiteSalary
💰 $272,000 - $306,000 per year
Job Level
Lead
Tech Stack
PythonPyTorchTensorflow
About the role
- Raise the technical bar for the team by setting architectural direction, tackling systemic challenges, and identifying opportunities for innovation and efficiency.
- Design, build and scale robust, high-throughput, low latency recommender systems that power product features used by tens of millions of users every day.
- Build and deploy advanced ML models, leveraging deep learning, reinforcement learning, and optimization to drive engagement and enhance user experience.
- Collaborate cross-functionally to shape ML-driven product roadmaps, balancing speed of iteration with long-term system complexity and scalability.
Requirements
- 8+ years of experience in applied Machine Learning, inclusive Ph.D. or Master’s degree in Computer Science, Machine Learning, Artificial Intelligence, or a related field.
- Deep expertise in mainstream RecSys model architecture (e.g. two-tower, transformer-based model, multi-task learning, wide and deep etc.).
- Strong proficiency in Python and ML frameworks such as PyTorch, JAX, or TensorFlow.
- Extensive experience building performant machine learning systems at scale and have driven execution from ideation to production implementation.
- Strong product intuition and a passion for building ML applications grounded in user feedback and real-world impact.
- Excellent communication and collaboration skills—able to lead complex, cross-functional technical initiatives and keep stakeholders aligned through clear updates and problem-solving.
- The ability to thrive in ambiguous environments, and are energized by tackling open-ended, technically challenging problems.
Benefits
- equity
- benefits 📊 Resume Score Upload your resume to see if it passes auto-rejection tools used by recruiters Check Resume Score
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard skills
Machine LearningDeep LearningReinforcement LearningOptimizationPythonPyTorchJAXTensorFlowRecommender SystemsModel Architecture
Soft skills
CommunicationCollaborationLeadershipProblem-solvingProduct IntuitionAdaptabilityStakeholder ManagementCross-functional CollaborationInnovationEfficiency
Certifications
Ph.D. in Computer ScienceMaster’s degree in Machine LearningMaster’s degree in Artificial Intelligence