AI Framework Engineer

Title: AI Framework Engineer

Company: webAI

Location: Grand Rapids, MI

Type: Full-Time, Salaried Exempt

Experience: 5-10 years

Education: Bachelor’s (minimum) or Master's Degree in Electrical/Computer Engineering, Computer Science, or other related Engineering discipline

About Us:

webAI is a software company that is building a decentralized AI development platform. Our technology enables the development of powerful AI using limited amounts of data, challenging the underlying assumption that big data is the key to unlocking the full power of AI. webAI offers opportunities to join a dynamic and fast-growing team that fosters a fun and growth-oriented office culture. We are seeking an experienced AI Framework Engineer to help spearhead the development of an innovative AI framework tailored for distributed and dynamic computing environments. As a key member of our team, you will work to design, implement, and optimize this framework, enabling seamless integration of AI algorithms and models across infrastructure managed by our platform.

We at webAI are committed to living out the core values we have put in place as the foundation on which we operate as a team. We seek individuals who exemplify the following:

• Truth - Emphasizing transparency and honesty in every interaction and decision.
• Ownership - Taking full responsibility for one’s actions and decisions, demonstrating commitment to the success of our clients.
• Tenacity - Persisting in the face of challenges and setbacks, continually striving for excellence and improvement. 
• Humility - Maintaining a respectful and learning-oriented mindset, acknowledging the strengths and contributions of others.

Responsibilities:

• Build, document, optimize, and maintain AI framework related software packages to enable successful training and deployment of models and algorithms across dynamic heterogeneous infrastructure

• Design and implement testing frameworks and methodologies via CI/CD to ensure framework reliability
• Collaborate closely with research, engineering, and product teams to understand requirements, define technical specifications, and prioritize framework features/attributes
• Stay updated on the latest developments in machine learning algorithms, frameworks, and technologies

Requirements and Skills:

• Strong Python and C++ (or Rust) programming skills
• Experience optimizing AI inference and training algorithms in the context of hardware accelerators (Apple Silicon, GPU, TPU, etc.)
• Deep understanding of AI algorithm architectures, frameworks (e.g., TensorFlow, PyTorch), and their integration into distributed environments
• Experience building, training, and deploying AI models to production environments
• Understanding of common federated learning patterns and approaches
• Strong problem-solving skills and ability to address complex technical challenges related to distributed systems, performance optimization, and resource management.
• Familiarity with containerization technologies (e.g., Docker, Kubernetes) and cloud platforms (e.g., AWS, Azure, Google Cloud) for deployment and orchestration.

Benefits:

• Competitive salary and performance-based incentives.
• Comprehensive health, dental, and vision benefits package.
• $200/mos Health and Wellness Stipend
• $400/year Continuing Education Credit
• Flexible work week
• Free parking, for in-office employees
• Unlimited PTO
• Parental, Bereavement Leave
• Supplemental Life Insurance

webAI is an Equal Opportunity Employer and does not discriminate against any employee or applicant on the basis of age, ancestry, color, family or medical care leave, gender identity or expression, genetic information, marital status, medical condition, national origin, physical or mental disability, political affiliation, protected veteran status, race, religion, sex (including pregnancy), sexual orientation, or any other characteristic protected by applicable laws, regulations and ordinances. We adhere to these principles in all aspects of employment, including recruitment, hiring, training, compensation, promotion, benefits, social and recreational programs, and discipline. In addition, it is the policy of webAI to provide reasonable accommodation to qualified employees who have protected disabilities to the extent required by applicable laws, regulations and ordinances where a particular employee works.