Let’s get started
By clicking ‘Next’, I agree to the Terms of Service
and Privacy Policy, and consent to receive emails from Rise
Jobs / Job page
Software Engineer - Model Training Infrastructure image - Rise Careers
Job details

Software Engineer - Model Training Infrastructure

About Anyscale:


At Anyscale, we're on a mission to democratize distributed computing and make it accessible to software developers of all skill levels. We’re commercializing Ray, a popular open-source project that's creating an ecosystem of libraries for scalable machine learning. Companies like OpenAIUberSpotifyInstacartCruise, and many more, have Ray in their tech stacks to accelerate the progress of AI applications out into the real world.


With Anyscale, we’re building the best place to run Ray, so that any developer or data scientist can scale an ML application from their laptop to the cluster without needing to be a distributed systems expert.


Proud to be backed by Andreessen Horowitz, NEA, and Addition with $250+ million raised to date.


About The Role

We’re looking for passionate, motivated engineers excited to build infrastructure and tools for the next generation of machine learning applications. We’re hiring exceptional Software Engineers for our distributed training team, which develops and maintains widely adopted open-source machine learning libraries.


We’re particularly interested in engineers who can help shape and execute a vision for the future of ML training infrastructure. We welcome both Individual Contributors and technically inclined individuals with experience managing small teams.


About The Distributed Training Team

The Distributed Training team drives the development and optimization of Ray’s distributed training libraries, focusing on features and performance enhancements for large-scale machine learning workloads. They are responsible for building and maintaining core libraries like Ray Train (for distributed model training) and Ray Tune (for distributed hyperparameter tuning). You’ll collaborate closely with the Ray Core and Ray Data teams to create impactful, end-to-end solutions, and have the exciting opportunity to work directly with Machine Learning teams around the globe, shaping products that are transforming the AI landscape.

As part of this role, you will:

  • Develop scalable, fault-tolerant distributed machine learning libraries that power leading ML platforms

  • Create an exceptional end-to-end experience for training machine learning models

  • Solve complex architectural challenges and transform them into practical solutions

  • Contribute to and engage with the open-source community, collaborating with ML researchers, engineers, and data scientists to build new scalable machine learning abstractions

  • Share your work and expertise with a broader audience through talks, tutorials, and blog posts

  • Collaborate with a team of experts in distributed systems and machine learning

  • Work directly with end-users to iterate on and enhance the product based on their feedback

  • Partner with engineering and product managers to nurture a talented team of software engineers

  • Play a key role in building and shaping a world-class company

We'd love to hear from you if have:

  • Minimum 5+ years of experience building, scaling, and maintaining software systems in production environments

  • Strong fundamentals in algorithms, data structures, and system design

  • Proficiency with machine learning frameworks and libraries (e.g., PyTorch, TensorFlow, XGBoost)

  • Experience designing fault-tolerant distributed systems

  • Solid architectural skills

Bonus points if:

  • Experience with cloud technologies (AWS, GCP, Kubernetes)

  • Hands-on experience building ML training platforms in production

  • Background in managing and maintaining open-source libraries

  • Experience leading small teams to achieve ambitious technical goals

  • Familiarity with Ray

Compensation:

  • At Anyscale, we take a market-based approach to compensation. We are data-driven, transparent, and consistent. The target salary for this role is $170,112 ~ $237,000. As the market data changes over time, the target salary for this role may be adjusted.

  • This role is also eligible to participate in Anyscale's Equity and Benefits offerings, including the following:

  • Stock Options

  • Healthcare plans, with premiums covered by Anyscale at 99%

  • 401k Retirement Plan

  • Education & Wellbeing Stipend

  • Paid Parental Leave

  • Fertility Benefits

  • Flexible Time Off

  • Commute reimbursement

  • 100% of in office meals covered

Anyscale Inc. is an Equal Opportunity Employer. Candidates are evaluated without regard to age, race, color, religion, sex, disability, national origin, sexual orientation, veteran status, or any other characteristic protected by federal or state law. 


Anyscale Inc. is an E-Verify company and you may review the Notice of E-Verify Participation and the Right to Work posters in English and Spanish

Average salary estimate

$203556 / YEARLY (est.)
min
max
$170112K
$237000K

If an employer mentions a salary or salary range on their job, we display it as an "Employer Estimate". If a job has no salary data, Rise displays an estimate if available.

Similar Jobs
Photo of the Rise User
Anyscale Hybrid No location specified
Posted 24 hours ago

Join Anyscale as an Account Executive - High Tech and help shape the future of distributed computing while engaging with top digital native clients.

Photo of the Rise User
Posted 9 days ago

Lead a dynamic engineering team at Visa in Warsaw, driving innovative solutions within a global payments environment.

Simplified IT Solutions Hybrid Los Angeles Ave, Los Angeles, CA 90063, USA
Posted 15 hours ago

Join Simplified IT as a Senior MuleSoft Architect and drive innovative integration solutions across diverse industries.

Photo of the Rise User
Pinterest Remote San Francisco, CA, US; Remote, US
Posted 10 days ago

Elevate your career at Pinterest as a Senior Staff Software Engineer, focusing on the ML Platform to empower innovative solutions.

Photo of the Rise User
Vanta Hybrid No location specified
Posted 4 days ago
Inclusive & Diverse
Growth & Learning
Customer-Centric
Collaboration over Competition
Medical Insurance
Maternity Leave
Flex-Friendly
401K Matching

Vanta is looking for a Senior Software Engineer to drive impactful product development and mentor fellow engineers in a remote setting.

Wantable Hybrid No location specified
Posted 5 days ago

Fluency is on the hunt for a skilled Senior Front End Developer to enhance user experiences across multiple platforms.

Photo of the Rise User
Customer-Centric
Mission Driven
Inclusive & Diverse
Rise from Within
Diversity of Opinions
Work/Life Harmony
Growth & Learning
Transparent & Candid
Medical Insurance
Paid Time-Off
Maternity Leave
Mental Health Resources
Equity
Child Care stipend
Paternity Leave
WFH Reimbursements
Flex-Friendly
Dental Insurance
Vision Insurance
Life insurance
Health Savings Account (HSA)
Flexible Spending Account (FSA)
401K Matching
Military leave

NVIDIA invites experienced software engineers to innovate in autonomous vehicle technology as part of our Planning and Controls team.

Photo of the Rise User
Apple Hybrid Seattle, Washington, United States
Posted 12 days ago
Inclusive & Diverse
Diversity of Opinions
Work/Life Harmony
Dare to be Different
Reward & Recognition
Empathetic
Take Risks
Growth & Learning
Transparent & Candid
Mission Driven
Passion for Exploration
Feedback Forward
Medical Insurance
Dental Insurance
Vision Insurance
Mental Health Resources
Life insurance
Disability Insurance
Health Savings Account (HSA)
Flexible Spending Account (FSA)
Learning & Development
Paid Time-Off
Maternity Leave
Social Gatherings

Join Apple Data Platform to lead in building and maintaining world-class ML infrastructure as a Sr. Software Engineer in Seattle.

Photo of the Rise User
Certn Remote No location specified
Posted 9 days ago
Dental Insurance
Family Medical Leave
Paid Holidays

Join Certn as a Staff Engineer to shape the future of background screening with innovative technology.

Photo of the Rise User
QODE Hybrid No location specified
Posted 8 days ago

A leading company is on the hunt for a Senior .NET Developer to create high-performance web applications in Dallas or Chicago.

Photo of the Rise User
Ontic Remote No location specified
Posted 8 days ago

Become part of Ontic as a Software Engineer to innovate security solutions for corporate and government clients.

Photo of the Rise User
Inclusive & Diverse
Diversity of Opinions
Collaboration over Competition
Growth & Learning
Transparent & Candid
Medical Insurance
Mental Health Resources
Learning & Development
Flex-Friendly

Grafana Labs is looking for a Senior Backend Engineer with GoLang expertise to elevate its backend services for a global user base.

Photo of the Rise User
Posted 8 days ago

Join Creative Market as a Senior Search Platform Engineer to lead impactful search functionality improvements in a fully remote environment.

Photo of the Rise User
Together AI Remote San Francisco, California, United States
Posted 12 days ago

Join Together AI as a Senior Software Engineer and help shape the future of AI cloud infrastructure.

We are building the future of software development.

50 jobs
MATCH
Calculating your matching score...
FUNDING
SENIORITY LEVEL REQUIREMENT
TEAM SIZE
EMPLOYMENT TYPE
Full-time, remote
DATE POSTED
April 3, 2025

Subscribe to Rise newsletter

Risa star 🔮 Hi, I'm Risa! Your AI
Career Copilot
Want to see a list of jobs tailored to
you, just ask me below!