Title: Optimization Engineer
Company: webAI, Inc.
Location: Austin, TX
Type: Full-Time, Salaried Exempt
Experience: 5+ years
Education: Bachelor’s Degree, minimum
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 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.
Role Overview
We're seeking an Optimization Engineer to join our team and drive the performance improvements of our AI systems. In this role, you'll work on optimizing AI models, runtime environments, and infrastructure to ensure maximum efficiency and minimal resource usage across various local deployment scenarios.
Key Responsibilities
- Analyze and optimize the performance of AI models for execution on local infrastructure
- Develop and implement algorithms for model compression, quantization, and pruning
- Optimize memory usage, computation speed, and energy efficiency of AI workloads
- Collaborate with runtime and systems teams to ensure end-to-end optimization
- Create benchmarking tools and performance metrics for continuous improvement
- Research and implement state-of-the-art optimization techniques from academic papers
- Contribute to the development of auto-optimization tools for our AI platform
Requirements
- 5+ years of experience in performance optimization, preferably in AI/ML systems
- Strong background in computer science, mathematics, or a related field
- Proficiency in Python and C++; familiarity with Rust or Go is a plus
- Deep understanding of machine learning algorithms and neural network architectures
- Experience with model compression techniques (quantization, pruning, distillation)
- Familiarity with hardware architectures (CPU, GPU, TPU) and their performance characteristics
- Strong analytical and problem-solving skills
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.