Ant Group traces its roots back to Alipay, which was established in 2004. Today, we are a world-leading internet open platform. Through technological innovation, we support our partners to enable inclusive, convenient digital life and digital financial services for consumers and SMEs (small and micro-sized enterprises). We offer a wide range of digital financial services, including digital payments and digital banking.
We are looking for self-motivated AI engineers to join Ant's International Business Group. With high standards we comply on data security and privacy protection, our AI team is leveraging privacy-preserving compuation technologies such as federated learning in an increasing range of business scenarios. We aim to establish a shared, collaborative, and mutually beneficial ecosystem with regulatory bodies and institutions from various countries as we continue to unleash the value of data by AI.
Responsibilities:
- Design, develop, and maintain an innovative federated learning framework
- Develop protocols and communication mechanisms for secure data aggregation and model updates from distributed devices
- Research and implement state-of-the-art Federated Learning algorithms, ensuring that they are suitable for various machine learning tasks and maintain data privacy
- Optimize algorithms for performance, convergence speed, and resource utilization, while considering the challenges posed by decentralized data sources
- Implement data preprocessing, model initialization, and aggregation mechanisms that ensure effective training across a heterogeneous set of devices
- Design and execute comprehensive testing strategies to verify the reliability, robustness, and accuracy of the Federated Learning framework
- Create clear and comprehensive documentation, including technical specifications, design documents, and usage guides
- Collaborate effectively with cross-functional teams including machine learning researchers, software developers, and product managers to align on project goals and requirements
Requirements:
- Master's in Computer Science, Electrical Engineering, or a related field
- PhD is a nice to have
- Possess 5+ years of experience in federated learning or distributed machine learning
- Strong programming skills in Python
- Experience with machine learning frameworks (TensorFlow, PyTorch, scikit-learn)
- Familiar with software configuration management tools such as Git or GitLab.
- Knowledge of Continuous Integration/Delivery/Deployment
- Strong understanding of federated learning principles and algorithms
- Strong communication skills
- Proficiency in English and Mandarin in order to communicate with our Mandarin-speaking stakeholders
