AI Infrastructure Engineer, Scale AI
Model Serving Platform
$175-220k
Plus equity
Training and validation data for AI applications
Be an early applicant
Training and validation data for AI applications
1001+ employees
Be an early applicant
$175-220k
Plus equity
1001+ employees
Company mission
To accelerate the development of AI applications.
Role
Who you are
- The ideal candidate is someone who has strong fundamentals in machine learning, backend system design, and has prior ML Infrastructure experience
- They should also be comfortable with infrastructure and large scale system design, as well as diagnosing both model performance and system failures
- 4+ years of experience developing ML platforms
- Passion for working closely with researchers to drive business impact
- Experience training and/or benchmarking LLMs
- Experience with Python, Docker, Kubernetes, and Infrastructure as code (e.g. terraform)
Desirable
- Experience building, deploying, and monitoring complex microservice architectures
- Experience working with a cloud technology stack (eg. AWS or GCP)
What the job involves
- As a software engineer on the ML Infrastructure team, you will work on developing the platform for orchestrating post-training and model evaluation jobs
- At Scale, we are constantly developing new data sources and running experiments to understand their impact on ML models
- To support this effort, we are looking for engineers who are comfortable navigating cloud infrastructure challenges as well as research challenges in benchmarking and tuning LLMs
- Develop re-usable platforms for running in-house and open-source LLM-benchmarks
- Ensure correctness and performance of post-training and eval jobs on the platform
- Improve APIs for managing ML workflows
- Contribute to foundational infrastructure at the company for model inference and training
- Participate in our team’s on call process to ensure the availability of our services
- Own projects end-to-end, from requirements, scoping, design, to implementation, in a highly collaborative and cross-functional environment
Our take
When training AI models, lack of access to large scale, high quality, labelled data is often the barrier preventing machine learning teams from building something great. In fact, labelled data is even more essential than the algorithms themselves. Scale AI is solving this by providing easy access through developer APIs to a huge variety of labelled data sets covering a large range of areas within computer vision and natural language processing.
It seems that almost every company now is offering artificial intelligence enabled products to automate every aspect of our lives. By positioning themselves as the go-to source for high quality tagged data, Scale AI is in an excellent position as the AI sector continues it's exponential growth.
With innovative AI projects across a range of sectors including retail, energy, and healthcare, Scale AI is now a leader in the industry alongside OpenAI. It continues to forge partnerships with fintech tools and software engine developers such as Brex and Luminar to continue its evolution of digital markets.
Freddie
Company Specialist at Welcome to the Jungle
Insights
Few candidates hear
back within 2 weeks
31% female employees
38% employee growth in 12 months
Company
Funding (last 2 of 7 rounds)
May 2024
$1bn
SERIES F
Apr 2021
$325m
SERIES E
Total funding: $1.6bn
Company benefits
- Health, Dental & Vision Coverage: Our health plans give you the flexibility to select the right coverage for you and eligible family members through a variety of plan options
- Easy to use 401(K): Plan and invest for the future with a 401(k) via Guideline. Scale’s 401(k) plan provides you an opportunity to defer compensation for your long-term savings
- Wellness Fund: We care about the physical, mental, and emotional wellbeing of all Scaliens. Our $100/month wellness stipend can be used for gym memberships, acupuncture, meditation apps, and so much more
- Virtual Social Activities: Being remote has not stopped us from hosting fun virtual events. From trivia night to candle making, we ensure employees are fostering connections & building strong relationships
- Learning & Development: We know how important career growth is for Scaliens, so we offer a $500/year L&D stipend to help support continued development throughout your journey
- Flexible Work Hours: Flexible hours allow you to work when you are most productive. You can work with your manager to best plan your daily work schedule
- Generous Paid Time Off: Enjoy time to travel or plan a staycation. We encourage employees to take time off to recharge and prevent burnout. We have a flexible PTO policy where each employee is afforded the flexibility to take planned time-off as needed
- Commuter Benefits: Set aside pre-tax dollars to use on qualified transportation expenses to help ease your commute
- Parental Leave: Balancing work and family is essential, and Scale understands the importance of having adequate leave policies in place to promote a healthy home and work life
Company values
- Earn customer love: We are passionate about our customers and devoted to their success. We know that customer trust and confidence is earned, not given, with every interaction, every delivery, and every time we solve a customer problem
- Intellectual rigor, open mind: We think in clear verifiable hypotheses and continuously obtain data that validates or refutes them. We ask questions and challenge assumptions to get consistently to the truth
- Ambition shapes reality: We are extremely ambitious in what we can accomplish. We set high standards, expecting to fail in the short-term but knowing that failure guides us to learn and ultimately succeed
- Team flow: We are energized by each other’s success and actively invest in others with care, candor, and focus. Ideas flow openly between us because we treat each other with trust and respect
- Ownership is the job: We think like owners. We embody expansive ownership and are big believers in our ability to drive outcomes and enact change
- Run through walls: We are undaunted by barriers and race towards overcoming them. We step outside of our comfort zone, test our ideas, and do the hard work to get to the right solution to a hard problem
- Up the tempo: We move with agility and intent, and swiftly make two-way door decisions that allow us to learn faster, adapt quickly, and ultimately do more
- Results speak loudest: We define goals with intent and evaluate our outcomes with objectivity. At the end of the day, we measure ourselves by our results
Company HQ
Rincon Hill, San Francisco, CA
Articles
Leadership
Alexander Wang
(CEO)Dropped out of a bachelors degree at MiT in Computer Science and Mathematics. Previously worked for 2 years as a Software Engineer and Algorithm Developer in the finance industry and 2 years as a Tech Lead at Quora. Youngest founder funded by Accel as a teenager.
Salary benchmarks
We don't have enough data yet to provide salary benchmarks for this role.
Submit your salary to help other candidates with crowdsourced salary estimates.
Share this job
View 89 more jobs at Scale AI