AI/ML Engineer

Job Title: AI/ML Engineer

We are seeking an AI/ML Engineer to develop innovative AI solutions for our cutting-edge technology. In this role, you will tackle complex machine learning challenges, contribute to groundbreaking research, and help redefine the future of medical diagnostics.

Responsibilities

• Develop and train multi-task ML models for stroke detection using RF-derived features, including binary and multi-class classification as well as auxiliary prediction tasks.
• Build generative models (GANs, diffusion models, VAEs) for diagnostic image generation and visualization.
• Optimize models for real-time inference and deployment constraints.
• Collaborate with DSP engineers to understand signal characteristics and refine feature pipelines.
• Evaluate model performance and iterate on architectures based on clinical validation data.
• Support integration of models into production medical device software.
• Participate in weekly technical meetings to showcase progress, pitch new ideas, and brainstorm solutions to existing issues.

Required Qualifications

• M.S. in Computer Science, Electrical Engineering, or related field, or 5+ years of industry experience in machine learning.
• Experience with generative models (diffusion models, VAEs, or encoder-decoder architectures) applied to 2D/3D data.
• Proficiency in Python and deep learning frameworks (PyTorch or TensorFlow).
• Strong foundation in algorithm optimization and data structures.

Preferred Qualifications

• Signal or image processing background, particularly with radar, microwave, or compressed sensing techniques.
• Experience with multimodal models or cross-modal generation.
• Prior work optimizing models for GPU or edge deployment.
• Familiarity with medical device regulatory standards and a clear understanding of the application of AI technologies within FDA-regulated environments.
• Experience in medical device or healthcare technology companies.

Compensation Structure

This role begins with a 3-6 month contract phase at $4,000-$7,000/month to establish mutual fit. Upon conversion to full-time employment, salary range is $100,000-$120,000 plus equity, based on experience. Full-time benefits include health insurance (medical, dental, vision) and performance bonuses.