The Machine Learning Engineer will be responsible for the end-to-end development and deployment of Large language and machine learning models, with a primary focus on data preprocessing, model training, and fine-tuning using large-scale healthcare datasets.
Requirements
- Bachelor's or Master's degree in Computer Science, Machine Learning, Artificial Intelligence, or a related quantitative field.
- 5+ years of experience in Machine Learning Engineering or a similar role.
- Proven experience with large-scale data preprocessing, LLM/model training, and fine-tuning.
- Experience with distributed training (PyTorch Distributed, DeepSpeed, Ray, Hugging Face Accelerate).
- Experience with GPU/TPU optimization, memory management for large language models.
- Proficiency in Python and relevant ML libraries (e.g., TensorFlow, PyTorch, Scikit-learn, Pandas, NumPy).
- Strong understanding of various machine learning algorithms,Large Language Models, and deep learning architectures.
- Experience with cloud platforms (e.g., GCP, AWS) and distributed computing frameworks (e.g., Spark) is a plus.
- Familiarity with MLOps practices and tools.
- Ability to work independently and as part of a team in a fast-paced environment.
- Excellent problem-solving and analytical skills.
- Strong communication and collaboration abilities.
- Work Authorization: Must be a US Citizen, Green Card holder, or currently in the US have valid H1B visa
Benefits
- Competitive salary and benefits package.
- Flexible working arrangements (remote or hybrid options available).
- The opportunity to work on life-changing AI technology that directly impacts patient outcomes.
- Join a team that combines cutting-edge innovation with a mission to save lives and improve health equity.
- Continuous learning opportunities with access to the latest tools and advancements in AI and healthcare.