TN Visa AI Software Developer Jobs

About Us: webAI 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.

About the Role: We are looking for an AI Software Engineer with deep expertise in high-performance computing and machine learning inference. The AI Engineer is the "builder" who makes the science real. They take raw models and turn them into scalable, efficient, and reliable software. They are the masters of the production environment, focusing on latency, reliability, and data integrity. This is an ideal opportunity for someone who thrives at the intersection of low-level systems programming, applied machine learning, and hardware-aware optimization. This position is based in-office at our downtown Austin headquarters.

Key Responsibilities

• Model Deployment: Wrap models in high-performance APIs and integrate them into the broader software ecosystem.
• Data Engineering: Build and maintain the ETL pipelines that feed the AI Scientist’s experiments and the production inference engines.
• Optimization: Implement techniques like pruning, quantization, and distillation to ensure models run efficiently on edge devices or in high-traffic cloud environments.
• System Testing: Develop robust testing suites for AI behaviors, including regression testing for "hallucinations" or bias.
• Collaboration: Work closely with the AI Scientist to refactor "research code" into production-grade, maintainable code.

Requirements & Skills

• Experience: 5+ years in Software Engineering or equivalent university time with a deep experience that matches WIL objectives to include with a focus on ML-specific systems and distributed computing.
• Technical Depth: Proficiency in C++/Python, Docker/Kubernetes, Vector Databases (Pinecone, Weaviate), and CI/CD for ML.
• Education: Bachelor’s or Master’s in Computer Science or Software Engineering.

Benefits:

• Competitive salary
• Equity options
• Comprehensive health, dental, and vision benefits package
• 401(k) match (U.S.-based employees only)
• $200/month Health & Wellness stipend
• Continuing Education support
• $500/year Function Health subscription (U.S.-based employees only)
• Free parking for in-office employees
• Flexible Time Off (FTO)
• Parental leave for eligible employees
• 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, 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.