About the Role
We are seeking a talented AI Engineer to join our team and help us develop and deploy cutting-edge AI solutions. As an AI Engineer, you will be responsible for designing, building, and maintaining AI models that will be used to improve our products and services. You will work closely with data scientists, software engineers, and other stakeholders to ensure that our AI solutions are aligned with our business goals.
Responsibilities:
- Design and implement petabyte-scale data processing systems
- Manage workloads across large cloud compute clusters
- Pre-process complex datasets for AI training
- Collaborate with the team to optimize data pipelines and model performance
You're a perfect fit if you have:
- Strong engineering skills and a passion for data and model improvement
- Experience working with non-textual data modalities (e.g., images, audio, video)
- A track record of building custom data processing libraries from scratch
- Expertise in designing and implementing distributed systems using Rust
- Up-to-date knowledge of state-of-the-art AI data preparation techniques
- Excellent organizational skills and meticulous data bookkeeping across multiple clouds and sources
Bonus points for:
- Experience with JAX, a high-performance numerical computation library for Python
- Experience with Apache Spark, a distributed processing framework
What we offer:
- Competitive salary range: $170,000 - $350,000 USD
- Opportunity to work on challenging and impactful projects at the forefront of AI
- Collaborative and supportive work environment in the heart of the Bay Area
- Streamlined interview process directly with our technical team