Remote AI Framework Engineer

Description:

• webAI is seeking an experienced AI Framework Engineer to join their team in Grand Rapids, MI, or remotely.
• The role involves designing, implementing, and optimizing an innovative AI framework for distributed and dynamic computing environments.
• Responsibilities include building, documenting, optimizing, and maintaining AI framework-related software packages for model training and deployment.
• The engineer will design and implement testing frameworks and methodologies via CI/CD to ensure framework reliability.
• Collaboration with research, engineering, and product teams is essential to understand requirements and prioritize framework features.
• Staying updated on the latest developments in machine learning algorithms, frameworks, and technologies is expected.

Requirements:

• A minimum of 5-10 years of experience in a related field is required.
• A Bachelor’s or Master's Degree in Electrical/Computer Engineering, Computer Science, or a related Engineering discipline is necessary.
• Strong programming skills in Python and C++ (or Rust) are essential.
• Experience in optimizing AI inference and training algorithms for hardware accelerators (Apple Silicon, GPU, TPU, etc.) is required.
• A deep understanding of AI algorithm architectures and frameworks (e.g., TensorFlow, PyTorch) and their integration into distributed environments is necessary.
• Experience in building, training, and deploying AI models to production environments is required.
• Familiarity with common federated learning patterns and approaches is essential.
• Strong problem-solving skills to address complex technical challenges related to distributed systems and performance optimization are necessary.
• Familiarity with containerization technologies (e.g., Docker, Kubernetes) and cloud platforms (e.g., AWS, Azure, Google Cloud) for deployment and orchestration is required.

Benefits:

• The position offers a competitive salary and performance-based incentives.
• A comprehensive health, dental, and vision benefits package is provided.
• Employees receive a $200/month Health and Wellness Stipend.
• A $400/year Continuing Education Credit is available for professional development.
• The company offers a flexible work week to accommodate employees' needs.
• Free parking is provided for in-office employees.
• Unlimited Paid Time Off (PTO) is available.
• Parental and Bereavement Leave are included in the benefits package.
• Supplemental Life Insurance is also offered to employees.