webAI is hiring a
AI Framework Engineer in United States

Summary

webAI is seeking an experienced AI Framework Engineer to help develop an innovative AI framework for their decentralized platform. The engineer will be responsible for building, optimizing, and maintaining software packages, designing testing frameworks, collaborating with various teams, and staying updated on machine learning advancements. webAI offers a competitive salary, comprehensive health benefits, a wellness stipend, continuing education credit, flexible work week, free parking, unlimited PTO, parental leave, supplemental life insurance, and is an equal opportunity employer.

Requirements

• 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

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

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