Salary
💰 $180,000 - $205,000 per year
Tech Stack
AWSPythonPyTorch
About the role
- Develop and deploy ML and LLM models for recommendation systems and personalized treatment strategies
- Work extensively with Python and ML libraries like PyTorch to build and optimize models
- Write clean, efficient, and production-ready code to support scalable ML applications
- Collaborate with engineers and product managers to integrate ML models into real-world systems
- Set up necessary ML infrastructure components to streamline model training, evaluation, and deployment
- Evaluate and implement cutting-edge ML tools and frameworks, keeping the company at the forefront of ML and LLM advancements
- Mentor engineers, sharing expertise in ML modeling, deployment, and best practices
- Actively participate in retrospectives and help drive continuous improvement in our process and culture
Requirements
- 3+ years of experience in Machine Learning and/or Engineering, with a strong emphasis on hands-on model development and coding
- Expert-level proficiency in Python and deep experience in ML frameworks like PyTorch
- Experience with building and deploying production ML models at scale
- Strong background in LLMs and NLP applications, including applications like summarization and chatbots
- Familiarity with ML deployment best practices, with the ability to bring models from research to production
- Experience with ML infrastructure tools (e.g., Databricks, MLFlow, AWS SageMaker) is a plus
- A Master’s degree in Computer Science, Machine Learning, or a related field (not strictly required)
- A collaborative mindset, strong problem-solving skills, and the ability to work cross-functionally