AI Research Scientist

AI Research ScientistwebAI is pioneering the future of artificial intelligence by establishing the first distributed AI infrastructure dedicated to personalized AI. We recognize the evolving demands of a data-driven society for scalability and flexibility, and we firmly believe that the future of AI lies in distributed processing at the edge, bringing computation closer to the source of data generation. Our mission is to build a future where a company's valuable data and intellectual property remain entirely private, enabling the deployment of large-scale AI models directly on standard consumer hardware without compromising the information embedded within those models. We are developing an end-to-end platform that is secure, scalable, and fully under the control of our users, empowering enterprises with AI that understands their unique business. We are a team driven by truth, ownership, tenacity, and humility, and we seek individuals who resonate with these core values and are passionate about shaping the next generation of AI.The AI Research Scientist will contribute to webAI's development of next-generation AI models and systems. In this role, you will design, train, evaluate, and optimize cutting-edge machine learning models including large language models, multimodal architectures, and on-device inference systems. You will work closely with research leadership, applied AI teams, and platform engineering to advance scientific discovery while ensuring that innovations translate into real-world impact.This is a hands-on research role for someone who loves experimentation, solving complex problems, and building AI that is powerful, efficient, and privacy-preserving.ResponsibilitiesDesign, train, and optimize machine learning models including LLMs, multimodal models, transformers, and diffusion architecturesConduct research on model efficiency, quantization, compression, and on-device deploymentPrototype novel model architectures, training methods, and inference strategies for distributed AIDevelop and evaluate benchmarks, datasets, and experimental frameworks to test model performanceCollaborate with engineering teams to integrate research findings into production systemsStay current on leading research in deep learning, generative AI, and distributed MLAnalyze experimental results and communicate insights clearly to technical and non-technical stakeholdersDocument research findings, contribute to internal papers, and present technical work across the organizationIdentify emerging technologies and propose research directions aligned with webAI's strategic prioritiesQualifications4+ years of experience (can be graduate research) in machine learning research, AI model development, or related fieldsStrong expertise in deep learning architectures including transformers, CNNs, RNNs, and diffusion modelsHands-on experience training and fine-tuning large-scale modelsProficiency in Python and ML frameworks such as PyTorch, TensorFlow, or JAXExperience building datasets, designing experiments, and validating ML model performanceDeep understanding of optimization techniques including quantization, distillation, pruning, and hardware-aware trainingStrong problem-solving skills and ability to work independently on complex research tasksEffective communication skills for presenting research findings to diverse audiencesBachelor's degree in Computer Science, Engineering, Mathematics, or a related fieldPreferred SkillsMaster's or PhD in Machine Learning, Computer Science, AI, or a related fieldExperience with distributed training, edge inference, or on-device MLResearch experience in generative AI, reinforcement learning, or multimodal learningFamiliarity with privacy-preserving ML techniques such as federated learningExperience contributing to academic publications, patents, or open-source ML projectsComfort operating in a fast-paced, high-growth startup environmentWe 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.BenefitsWe strive to provide competitive benefits to all employees. The benefits listed in this posting generally apply to U.S.-based employees. For employees hired outside the United States, benefits may vary based on local law, country-specific requirements, and the employment platform or entity through which the employee is hired.Competitive salary and performance-based incentivesComprehensive health, dental, and vision benefits package including a high-deductible HSA option with company contribution401k match (US-based only)Flexible PTO PolicyParental leave for eligible employeesContinuing education supportBasic, company-paid life insuranceSupplemental life insuranceCompany-paid short and long term disabilityFree parking for in-office employeeswebAI 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, 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.