Machine Learning Engineer

Federated learning (FL) is a collaborative learning approach where multiple data owners, referred to as clients, train a model together under the orchestration of a central server by sharing the model trained on their local datasets instead of sharing the data directly. FL enables creation of more robust models without the exposure of local datasets. However, FL by itself, does not guarantee the privacy of data, because the information extracted from the communication of FL algorithms can be accumulated and utilized to infer the private local data used for training.

We developed Argonne Privacy Preserving Federated Learning framework (APPFL), with advances in differential privacy, to enable Privacy-Preserving Federated Learning (PPFL). We enabled training of AI models in a distributed setting across multiple institutions, where sensitive data are located, with the ability to scale on supercomputing resources to help create robust, trust-worthy AI models in biomedicine and smart grid applications where data privacy is essential.

Setting up a secure federated learning experiment that needs high performance computational resources across distributed sites requires technical capabilities that may not be available for all. To lower the barrier to entry for leveraging PPFL and to enable domain experts in large institutions to utilize FL, we created the Argonne Privacy-Preserving Federated Learning as a service (APPFLx), which enables cross-silo PPFL using easy to use web interface for managing, deploying, analyzing, and visualizing PPFL experiments. 

Successful candidates will have experience in developing FL frameworks, developing benchmarks for FL frameworks, fine-tuning LLMs using FL frameworks.

Position Requirements

• Working experience in Privacy preserving Federated Learning frameworks
• Bachelor’s or Master’s degree in Computer Science, Machine Learning, Mathematics, or a related field, and 0+ years of experience.
• Strong programming skills in Python and experience with relevant Machine Learning frameworks (e.g., TensorFlow, PyTorch).
• Solid understanding of machine learning concepts, algorithms, and techniques, particularly in the areas of federated learning, distributed optimization, and privacy-preserving techniques.
• Experience with cryptographic protocols, secure multi-party computation, and homomorphic encryption techniques is highly desirable.
• Familiarity with data privacy regulations and compliance requirements (e.g., GDPR, HIPAA).
• Excellent problem-solving, analytical, and critical thinking skills.
• Strong communication and collaboration skills, with the ability to work effectively in a cross-functional team environment.
• Proficiency in writing clean, maintainable, and well-documented code.
• Proven ability to learn and adapt to new technologies and methodologies.
• Ability to model Argonne’s core values of impact, respect, safety, integrity, and teamwork.

Job Family

Research Development (RD)

Job Profile

Software Engineering 1

Worker Type

Regular

Time Type

Full time

As an equal employment opportunity and affirmative action employer, and in accordance with our core values of impact, safety, respect, integrity and teamwork, Argonne National Laboratory is committed to a diverse and inclusive workplace that fosters collaborative scientific discovery and innovation. In support of this commitment, Argonne encourages minorities, women, veterans and individuals with disabilities to apply for employment. Argonne considers all qualified applicants for employment without regard to age, ancestry, citizenship status, color, disability, gender, gender identity, gender expression, genetic information, marital status, national origin, pregnancy, race, religion, sexual orientation, veteran status or any other characteristic protected by law.

Argonne employees, and certain guest researchers and contractors, are subject to particular restrictions related to participation in Foreign Government Sponsored or Affiliated Activities, as defined and detailed in United States Department of Energy Order 486.1A. You will be asked to disclose any such participation in the application phase for review by Argonne’s Legal Department.  

All Argonne offers of employment are contingent upon a background check that includes an assessment of criminal conviction history conducted on an individualized and case-by-case basis.  Please be advised that Argonne positions require upon hire (or may require in the future) for the individual be to obtain a government access authorization that involves additional background check requirements.  Failure to obtain or maintain such government access authorization could result in the withdrawal of a job offer or future termination of employment.