ML Platform Engineer

K Health is seeking a motivated and experienced machine learning engineer to join its team. As a machine learning engineer, you will work closely with data scientists, MDs, and software developers to develop, train, and serve models, as well as building AI systems that are changing the health industry sphere.

Requirements

• 2+ years of experience designing and implementing machine learning models
• 3+ years of industry experience with Python
• Experience with transformers, torch, CUDA, and other low-level ML libraries
• Familiarity with inference optimization techniques (SGLang/TensorRT/VLLM)
• Familiarity with tuning frameworks and libraries (Torchtune/trl/unsloth)
• Familiarity with relevant Machine learning algorithms
• Experience with cloud-based machine learning platforms such as Google cloud, Azure, or AWS
• Knack for code quality and optimization
• Strong problem-solving skills and ability to work independently and as part of a team
• Experience with ML workflow frameworks such as Kubeflow Vertex pipelines, AirFlow or Sagemaker pipelines (Nice to have!)

Benefits

• 20 paid vacation days
• 18 days sick leave
• 10 3-day weekends
• Hybrid work schedule with team meals and stocked fridges
• Commuter Benefits
• Community focused events
• Pension Plan
• Stipend per day for food
• Stock options for every full-time employee
• Vocational Studies Fund