Participate in and manage the research project with Principal Investigator (PI), Co-PI and the research team members to ensure all project deliverables are met.
Engage in forward-looking exploration and research of foundational AI technologies, including but not limited to large model pre-training and post-training paradigm innovation exploration, architecture design for multimodal understanding and generative models, next-generation foundational architectures (e.g., Transformer-next), and acceleration of large-scale model training and inference.
Lead the technical architecture and end-to-end development of AI Agentic products for a core enterprise platform, partnering closely with the Product Manager to shape product direction from a technical feasibility standpoint.
Performance Profiling & Optimization: Utilize profiling tools (e.g., Nsight, PyTorch Profiler) to identify bottlenecks in data loading, gradient computation, and communication. Implement optimizations like kernel fusion, sharding, and tiling to improve step time.
Collaborate with cross-functional teams to design, develop, and implement AI features, focusing on areas like natural language processing, data analysis, and predictive modeling.
Collaborate with cross-functional teams to design, develop, and implement AI features, focusing on areas like natural language processing, data analysis, and predictive modeling.