Machine Learning Backend Engineer
Machine Learning Backend Engineer
KOS is a HealthTech AI research lab developing wearable medical devices that monitor metabolic and cardiovascular health noninvasively. Our products are validated through clinical trials with leading academic medical centers and CROs, and we are on the regulatory pathway with the FDA. We are based at Stanford Research Park. As our clinical and product programs expand at an accelerating pace, we are growing the team across every active program.
About KOS: www.kos-ai.com
This role builds the backend that carries our models from training to patients. That covers the data pipelines for clinical trial data, the training and evaluation infrastructure on our own GPU cluster, and the services that deliver models to our wearable devices and mobile apps.
What you'll do
Build and run data pipelines for continuous biosignal data from active clinical trials.
Build model serving and inference services with clear latency and reliability targets.
Own the training infrastructure, including job scheduling, experiment tracking and dataset versioning.
Design the APIs and databases behind our clinical and consumer apps.
Build audit trails and data integrity checks fit for a regulatory submission.
Work with our research, firmware and clinical teams.
Qualifications
Strong Python. Production experience with backend services, PostgreSQL and Linux. Hands on experience deploying PyTorch models and running GPU workloads. You have built and operated a system that other people depend on every day.
Bonus: Time series or sensor data. Healthcare data security. Edge or on device inference. Distributed training.
Education: BS, MS or PhD in Computer Science, Electrical Engineering, or a related field.
What this role is not: Not a management role. Not remote. Not for someone whose hands on skills have faded. The bar is high.
Salary: $140,000 to $350,000 per year. Final offer depends on the result of the KOSQuest technical assessment.
Hiring process: Apply at kosquest.com. The assessment is an 8 hour trial day in person at our Palo Alto lab, with lunch provided. Strong results lead to a one hour specialization test and a same day offer.
Benefits: Meaningful early stage equity. KOS sponsors visas for every hire. Health, dental, vision, 401(k), unlimited PTO, and an unlimited compute budget.
Equal opportunity: KOS Inc. is an equal opportunity employer. Hiring decisions never depend on race, color, religion, sex, gender identity, sexual orientation, national origin, age, disability, veteran status, or similar traits. We hire on demonstrated skill, measured by the KOSQuest assessment. Candidates who need an accommodation for the trial day can ask when they are invited.