AI Architect

Job Description

• Fine-tuning state-of-the-art small language models (SLMs) like Llama-3.1-8B or Phi-3.5-3B on diverse healthcare datasets.
• Evaluating and optimizing various inference and serving frameworks (e.g., Triton, TensorRT, VLLM, TGI).
• Collaborating with product teams to review and refine architecture designs incorporating AI.
• Troubleshooting AI deployment issues in customer environments, ensuring smooth operation and optimal performance.
• Authoring and presenting technical documentation and white papers and showcasing breakthroughs in internal and external forums.

• Master’s or Ph.D. in Computer Science, Artificial Intelligence, Machine Learning, or a related field.
• Minimum of 7 years of hands-on experience in AI/ML, with a demonstrable track record of training and deploying LLMs and other machine learning models.
• Strong proficiency in Python and familiarity with popular AI/ML frameworks (TensorFlow, PyTorch, Hugging Face Transformers, etc.).
• Practical experience deploying and managing AI models in production environments, including expertise in serving and inference frameworks (Triton, TensorRT, VLLM, TGI, etc.).
• Proven experience in healthcare AI applications, including a solid understanding of healthcare data standards (FHIR, HL7) and regulatory compliance (HIPAA, SOC2, NIST).
• Excellent problem-solving and analytical abilities, capable of tackling complex challenges and evaluating multiple factors.
• Exceptional communication and collaboration skills, enabling effective teamwork in a dynamic environment.

• Generous PTO Benefits: Enjoy a PTO benefit accrual of 22 days per year.
• Parental Leave: Experience one of the industry's best parental leave policies.
• Rewards & Recognition: Monetary incentives and widespread recognition.
• Insurance Benefits: Medical, dental, and vision benefits, along with 100% company-sponsored short and long-term disability and basic life insurance.