Software Engineer MLX

webAI; Austin (TX); On-site; USD 100,000 - 140,000; Full time

Job summary

Join an innovative firm at the forefront of AI technology as an AI Software Engineer. In this exciting role, you'll transform state-of-the-art machine learning models into production-ready systems. Collaborate with talented teams to ensure high performance and reliability while pushing the boundaries of AI infrastructure. This position offers a unique opportunity to work on cutting-edge projects that redefine data processing and privacy. If you're passionate about AI and eager to make a significant impact in a rapidly evolving field, this is the perfect place for you to thrive.

Benefits

Comprehensive health, dental, and vision benefits

401k Match

$200/month Health and Wellness Stipend

$400/year Continuing Education Credit

Free parking for in-office employees

Unlimited Approved PTO

Supplemental Life Insurance

Qualifications

• 5+ years of experience in software engineering focused on AI/ML.
• Strong proficiency in C++ and Python for software development.
• Familiarity with AI/ML concepts and architectures.

Responsibilities

• Build and maintain AI-related software packages and services.
• Design efficient, production-ready AI/ML systems.
• Collaborate with teams to integrate AI algorithms into services.

Skills

C++

Python

AI/ML

Algorithms

Distributed Systems

CUDA Programming

Education

Bachelor's in Computer Science

Master's in Computer Science

Tools

PyTorch

CMake

MLOps

Job description

We are establishing the first distributed Al infrastructure dedicated to personalized Al.
The evolving needs of a data-driven society are demanding scalability and flexibility. We believe that the future of Al is distributed and enables real-time data processing at the edge, closer to where data is generated. We are building a future where a company's data and IP remains private and it's possible to bring large models directly to consumer hardware without removing information from the model.

About the Role

We're seeking an experienced AI Software Engineer to join our engineering team. As an AI Software Engineer at our company, you'll be at the intersection of cutting-edge AI research and software engineering. You'll work on transforming state-of-the-art ML models into highly efficient, production-ready systems that can operate at scale. This role combines the precision of systems engineering with the innovation of AI development, requiring both strong software engineering fundamentals and ML expertise. You'll collaborate with our research teams to implement novel AI architectures while ensuring our systems maintain high performance, reliability, and efficiency. This is an opportunity to shape the future of AI infrastructure while working on problems that push the boundaries of what's possible in machine learning.

Key Responsibilities:

• Build, document, optimize, and maintain (through CI/CD and testing) AI related software packages and services
• Design and implement efficient, production-ready AI/ML systems
• Collaborate closely with research, engineering, and product teams to understand requirements
• Integrate and optimize AI/ML algorithms and research derived by R&D into platform packages and services
• Debug complex issues within our AI stack

Qualifications:

• Bachelor’s or Master’s degree in Computer Science, Engineering, Mathematics, or a related field
• Minimum of 5 years of experience in software engineering with a focus on AI/ML.
• Strong proficiency in C++ and Python for software development
• Experience binding C/C++ into Python (binding in other languages a bonus)
• Proficiency with C++ build systems like CMake
• Proven experience with PyTorch; familiarity with MLX is a plus
• Strong understanding of computer science fundamentals and algorithms
• Familiarity with AI/ML concepts, including LLM and computer vision model architectures
• Familiarity with distributed systems and parallel computing concepts

Preferred Skills and Experience:

• Experience with CUDA programming and GPU driven compute acceleration
• Knowledge of ML model quantization and optimization techniques
• Familiarity with MLOps tooling and practices
• Contributions to open-source ML projects
• Experience developing software within distributed systems
• Experience contributing to AI frameworks

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.
• Competitive salary and performance-based incentives.
• Comprehensive health, dental, and vision benefits package.
• 401k Match
• $200/mos Health and Wellness Stipend
• $400/year Continuing Education Credit
• Free parking, for in-office employees
• Unlimited Approved PTO
• 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.

Create a Job Alert

Interested in building your career at webAI? Get future opportunities sent straight to your email.

Apply for this job

*

indicates a required field

First Name *

Last Name *

Email *

Phone *

Resume/CV *

Enter manually

Accepted file types: pdf, doc, docx, txt, rtf

Enter manually

Accepted file types: pdf, doc, docx, txt, rtf

Education

School * Select...

Degree * Select...

Select...

Start date year

End date month Select...

End date year

Are you legally authorized to work in the United States?Note: If hired, a Form I-9, Employment Eligibility Verification, must be completed at the start of employment. * Select...

Do you now, or will you in the future require sponsorship for employment (e.g;- H-1 B visa, Work Permit, etc) in the United States? * Select...

This role requires someone on-site at our Austin, Texas Corporate office. Are you within commutable distance to Austin, Texas? If not within commutable distance, Are you open to relocation to the Austin, Texas area? *

Do you have strong proficiency in C++ software development? * Select...

Do you have 5+ years of experience in Software Engineering with a focus on embedded systems or AI/ML. *

Do you have proven experience with PyTorch; familiarity with MLX is a plus? *

Do you have proficiency with C++ build systems like CMake? *

What problems are you passionate about solving using AI and what potential problems can AI cause? *