AI & Machine Learning Developer

AI and Machine Learning Developer role at Johnson Health Tech Companies in Cottage Grove, WI onsite, offering a salary of USD 99,000 - 119,000 per year.

Responsibilities

• Design, prototype, and implement AI and ML features for Johnson Health Tech mobile fitness apps.
• Develop and refine large language models to support natural language interactions and user personalization.
• Build scalable backend services leveraging AWS, including Lambda, DynamoDB, and SageMaker.
• Establish production ready ML operations pipelines and inference endpoints using AWS tools such as SageMaker and Bedrock.
• Collaborate with Android developers to integrate AI capabilities into client-side applications.
• Create prototypes and proof-of-concepts for new AI driven features.
• Stay current with emerging AI and ML technologies and best practices.
• Ensure compliance with data privacy and security standards.

Requirements

• Bachelor’s or Master’s degree in Computer Science, Data Science, or a related field.
• 3+ years of AI/ML development experience, with emphasis on Natural Language Processing and LLMs.
• Hands-on experience deploying AI/ML workloads on AWS.
• Strong Python skills and familiarity with ML frameworks such as TensorFlow and PyTorch.
• Experience with RESTful APIs and a microservices architecture.

Technologies

• Python
• TensorFlow
• PyTorch
• AWS Lambda
• DynamoDB
• SageMaker
• Bedrock
• Android

Benefits

• Health & Dental Insurance
• Company paid Life Insurance
• 401(k)
• Paid Time Off benefits
• Product discounts
• Wellness programs