AI & Machine Learning Developer

Position Overview

Under the direction of the Sr. Director of Electrical Engineering, the AI/ML Developer for Mobile Fitness Applications will design and prototype advanced artificial intelligence features for Johnson Health Tech’s mobile fitness applications. This role focuses on leveraging Large Language Models (LLMs) and AWS backend services to create innovative, personalized user experiences. The developer will collaborate closely with client-side Android developers to integrate these features into production applications.

Location: Cottage Grove, WI 53527

Responsibilities

• Research, design, and implement AI/ML solutions for mobile fitness applications.
• Develop and fine-tune LLMs for natural language interactions and personalization.
• Build scalable backend services using AWS technologies (Lambda, DynamoDB, SageMaker, etc.).
• Build scalable production ready ML ops pipeline and inference endpoints using AWS technologies e.g. SageMaker, Bedrock.
• Collaborate with Android developers to integrate AI features into client-side applications.
• Create prototypes and proof-of-concepts for new AI-driven features.
• Stay current with emerging AI/ML technologies and best practices.
• Ensure compliance with data privacy and security standards.

Requirements

• Education: Bachelor’s or Master’s degree in Computer Science, Data Science, or a related field.
• Experience: 3+ years of experience in AI/ML development, with a focus on Natural Language Processing (NLP) and LLMs.
• Hands-on experience with AWS services for AI/ML deployment.
• Proficiency in Python and ML frameworks (TensorFlow, PyTorch).
• Experience with RESTful APIs and microservices architecture.

Benefits

We offer a competitive compensation package and a collaborative, growth-oriented environment. Some of the benefits include:

• Health & Dental Insurance
• Company paid Life Insurance
• 401(k)
• Paid Time Off benefits
• Product discounts
• Wellness programs
• EOE/M/W/Vet/Disability
• #ZR