Key Responsibilities:
1.Design and develop AI systems(i.e. training acceleration framework, inference speedup framework or integration of framework) that meet business requirements, including natural language processing, computer vision, machine learning, and deep learning algorithms.
2. Lead the AI project lifecycle from conception to deployment, including data collection, preprocessing, model training, validation, and deployment.
3.Collaborate with cross-functional teams, including software engineers, data scientists, and product managers, to integrate AI solutions into existing products and services.
4.Stay up-to-date with the latest AI technologies and research, and evaluate their potential impact on our business and industry.
5.Ensure that all AI systems are secure, reliable, and scalable, and comply with regulatory requirements.
Requirements:
1. Proficient in TensorFlow/PyTorch, Ray/DeepSpeed/NVIDIA Megatron/Colossal-AI, and familiar with the internal operating mechanisms of these systems.
2. Familiar with various optimization algorithms and model architectures, proficient in Python or C++ optimization algorithm libraries, including various classical algorithms and models based on gradients (BERT, GPT-3, Swin Transformer, ViT, MLP-Mixer).
3. Familiar with SaaS, architecture, compilers, networks, CUDA, and other related knowledge or relevant project experience.
4. Strong programming and engineering implementation abilities. Those who have won programming competition awards or published high-quality papers will be given priority in recruitment.
5. Bachelor’s or Master’s degree in Computer Science, Artificial Intelligence, Machine Learning, or related field.
