Job – Researcher for Federated/Distributed Machine Learning/Systems


  • Participate in the development of machine learning platform and open source communities
  • Responsible for the foundational research and product development, and continuously improve the R&D efficiency
  • Responsible for feature development, algorithm optimization of the platform, improving user experience and usability through cutting-edge or mature technologies
  • Participate in or lead design reviews with peers and stakeholders to decide amongst available technologies; 
  • Review code developed by other developers and provide feedback to ensure best practices (e.g., style guidelines, checking code in, accuracy, testability, and efficiency).
  • Contribute to existing documentation or educational content and adapt content based on product/program updates and user feedback.

Minimum qualifications

  • Bachelor’s degree or equivalent practical experience in computer science or related areas.
  • 2 years of experience with software development in one or more programming languages (Python, Java, JavaScript, C/C++), or 1 year of experience with an advanced degree.
  • 2 years of experience with data structures or algorithms in either an academic or industry setting.
  • Good communication and writing skills in English environment.

Preferred qualifications

  • Mature research skills in machine learning or distributed/federated machine learning; be familiar with PyTorch or TensorFlow.
  • Developing new algorithms or system-algorithm co-design for federated or distributed training platform
  • Solve research challenges such as statistical or system heterogeneity, security, privacy for federated learning or distributed training platform
  • Have at least two related paper published at the following top-tier machine learning or system conferences:  

[Machine Learning] ICML, NeurIPS, ICLR; 


[Distributed/Cloud/Mobile/Database Systems] NSDI, OSDI, SOSP, SC, VLDB, MobSys, MobiCom, Sensys, HotOS, EuroSys, ICDCS, etc.

  • Good communication skills with software engineers to land the developed algorithms into real-world system.

About the job

FedML, Inc. ( aims to provide an end-to-end machine learning operating system for people or organizations to transform their data to intelligence with minimum efforts. FedML stands for “Fundamental Ecosystem Development/Design for Machine Learning” in a broad scope, and “Federated Machine Learning” in a narrow scope. At the current stage, FedML is developing and maintaining a machine learning platform that enables zero-code, lightweight, cross-platform, and provably secure federated learning and analytics. It enables machine learning from decentralized data at various users/silos/edge nodes, without the need to centralize any data to the cloud, hence providing maximum privacy and efficiency. It consists of a lightweight and cross-platform Edge AI SDK that is deployable over edge GPUs, smartphones, and IoT devices. Furthermore, it also provides a user-friendly MLOps platform to simplify decentralized machine learning and real-world deployment. FedML supports vertical solutions across a broad range of industries (healthcare, finance, insurance, smart cities, IoT, etc.) and applications (computer vision, natural language processing, data mining, and time-series forecasting). Its core technology is backed by more than 3 years of cutting-edge research of its co-founders, who are recognized leaders in the federated machine learning community. Recently, FedML has raised around 2 million USD to scale up the product and engineering team.

FedML’s researchers and software engineers develop the next-generation platform for machine learning and artificial intelligence. We’re looking for researchers or engineers who bring fresh ideas from all areas, including machine learning and its applications in compute vision, natural language processing, data mining, as well as large-scale system design and implementation for distributed/cloud computing/systems, security/privacy, mobile/IoT systems, networking, Web UI design and development. As a software engineer, you will work on a specific project critical to our customers’ needs. We hope our engineers to be a faster learner and be enthusiastic to tackle new problems as we continue to push technology forward. You will design, develop, test, deploy, maintain, and enhance software solutions.


CEO – Salman Avestimehr 

Salman Avestimehr is a world-renowned expert in federated learning with more than 20 years of R&D leadership in both academia and industry. He has been a Dean’s Professor and the inaugural director of the USC-Amazon Center on Trustworthy Machine Learning at University of Southern California. He has also been an Amazon Scholar in Amazon. He is a United States Presidential award winner for his profound contributions in information technology, and a Fellow of IEEE.

CTO – Chaoyang He 

Chaoyang He is PhD from the CS department at the University of Southern California, Los Angeles, USA. He has research experience on distributed/federated machine learning algorithms, systems, and applications, and published papers at top-tier conferences such as ICML, NeurIPS, CVPR, ICLR, AAAI, and MLSys. He also has rich experience in industry in the areas of distributed/cloud computing and mobile/IoT systems. He was an R&D Team Manager and Principal Software Engineer at Tencent, and also worked as researcher/engineer at Google, Facebook, Amazon, Baidu, and Huawei. He has received a number of awards in academia and industry. Homepage


We prefer California, but we also support office in other states or work from home if necessary.