Software Engineer (Ray Data)

Anyscale Remote
software engineer data data datasets team ai pipeline training data processing production arrow infrastructure
November 20, 2022
CA, San Francisco

Ray aims to provide a universal API for building distributed applications (e.g. a machine learning pipeline of feature engineering, model training, and evaluation). Data is usually a core element connecting these different stages, and therefore plays a critical role in Ray’s usability, performance, and stability. We are looking for strong engineers to build, optimize, and scale Ray’s Datasets library and data processing capabilities in general.


About the Ray Data team:


The Ray Data team currently develops and maintains the Ray Datasets library, which is already powering critical production use cases (e.g. large scale data compaction at Amazon, and ML pipeline at Alibaba). Ray Datasets is a Python library built on top of Apache Arrow and Ray Core (Ray’s C++ backend), and the Ray Data team interacts closely with Ray Core components including the scheduler and the memory & I/O subsystems. The Ray Data team also works closely with Ray’s ML libraries including Train, RLlib, and Serve.


A snapshot of projects you will work on:

- Performance of Ray Datasets at large scale (leveraging Arrow primitives, optimizing Ray object manager, etc.)

- Integration with ML training and data sources

- Stability and stress testing infrastructure

- Lead future work integrating streaming workloads into Ray such as Beam on Ray

- Differentiate Data operations in Anyscale hosted Ray service

As part of this role, you will:

    • Develop high quality open source software to simplify distributed programming (Ray)

    • Identify, implement, and evaluate architectural improvements to Ray core and Datasets

    • Improve the testing process for Ray to make releases as smooth as possible

    • Communicate your work to a broader audience through talks, tutorials, and blog posts

We'd love to hear from you if have:

    • At least 2 year of relevant work experience

    • Solid background in algorithms, data structures, system design

    • Experience in building scalable and fault-tolerant distributed systems

    • Experience with data processing, database internals including Spark or Dask (streaming is a plus)

Anyscale, the company founded by the creators of Ray, is accelerating the development and productionization of AI applications, at any scale. Ray, the fastest growing open source project in distributed AI, lets developers instantly scale applications from a laptop to the cloud, without the cost or expertise of building complex infrastructure. By empowering developers of all skill levels to effortlessly develop and scale the end-to-end AI lifecycle on a single compute substrate and deploy the app into production, Anyscale is enabling organizations to bring AI products to market faster and at lower costs. Backed by Andreessen Horowitz, NEA and Addition, Anyscale is based in San Francisco, CA.


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.

Report this job

Similar developer/software engineer jobs in san francisco ca