Senior Software Engineer – AI/ML

Job Description

Senior Software Engineer – AI/ML | InnovateAI Labs

The Tone:
This is a full-time hybrid role at InnovateAI Labs, located in San Francisco, CA. InnovateAI Labs is a pioneering technology company focused on revolutionizing healthcare through advanced Artificial Intelligence and Machine Learning solutions. The company’s mission is to empower healthcare providers and improve patient outcomes by developing intelligent platforms for enhanced diagnostics, personalized treatments, and streamlined operations. This pivotal role allows you to contribute to high-impact projects from initial concept through to production, directly influencing the future of healthcare technology and making a significant difference in patient care.

The TL;DR
• Role: Full Time
• Type: Full-time
• Location: San Francisco, CA (Hybrid)

• Team: Cutting-edge R&D team
• Mission: Design, develop, and deploy sophisticated AI models and scalable machine learning systems that power core healthcare products.
• Tech Stack: Python, TensorFlow, PyTorch, JAX, AWS, GCP, Azure, Spark, Kafka, Dask, Docker, Kubernetes

What You’ll Actually Do
• Lead: Lead the design, development, and implementation of robust, scalable, and efficient AI/ML algorithms and systems for healthcare applications.
• Collaborate: Collaborate closely with product managers, data scientists, medical experts, and other engineering teams to define requirements and deliver innovative solutions.
• Research: Research, evaluate, and integrate state-of-the-art AI/ML techniques (e.g., deep learning for image analysis, natural language processing for clinical notes) to improve existing products and create new capabilities.
• Develop: Develop and maintain production-grade machine learning pipelines, ensuring data quality, model performance, and system reliability in a regulated environment.
• Mentor: Mentor junior engineers, fostering their growth and sharing expertise within the team.

The Must-Haves
• Background: Bachelor’s or Master’s degree in Computer Science, Artificial Intelligence, Machine Learning, or a closely related quantitative field.
• Experience: 5+ years of professional experience in software development with a significant focus on AI/ML applications. Proven experience with building, deploying, and maintaining ML models in production environments, ideally within a regulated industry.
• Skills: Strong proficiency in Python and extensive experience with major AI/ML frameworks such as TensorFlow, PyTorch, or JAX. Solid theoretical and practical understanding of machine learning principles, algorithms (e.g., deep learning, CNNs, RNNs, transformers, statistical modeling), and data science methodologies. Familiarity with cloud platforms (AWS, GCP, or Azure) and MLOps practices for reliable model deployment and management. Excellent problem-solving skills, intellectual curiosity, and the ability to work effectively both independently and as part of a collaborative team. Strong communication and interpersonal skills, with the ability to articulate complex technical concepts to diverse audiences.
• Bonus: PhD in a relevant field (e.g., AI, ML, Biomedical Engineering). Experience with distributed systems and large-scale data processing technologies (e.g., Spark, Kafka, Dask). Prior experience in the healthcare domain, working with clinical data (EHR, medical imaging), and understanding of healthcare regulations (e.g., HIPAA). Proven track record of publishing research or contributing to open-source projects in AI/ML. Experience with containerization (Docker, Kubernetes) for microservices and model deployment.