AI Scientist/Research Scientist

Responsibilities:

• Model training for LLM and multimodal LLM, with focus on supervised fine-tuning and reinforcement learning from human feedback.
• Publish papers on top-tier conferences and journals, such as Nature Communication, New England, Lancet, ICML, NeurIPS, ICLR, KDD, SIGIR, AAAI, IJCAI, et al.
• Research and develop large language model powered product and systems, primarily within the mental health related area.
• Work with data scientist/engineer to extract and integrate data from various sources (e.g., cooperated hospitals) for model training.
• Improve efficiency and stability for large scale distributed training jobs
• Plan and lead the development of new and advanced data analytic techniques, methodologies and analytical solutions from design, prototyping, and testing.
• Conduct testing on data and AI models, interprets findings from testing, and evaluates model performance for scaling and deployment.
• Work in a team setting and apply proficient in statistics, scripting and programming languages required by the firm.
• Work with relevant software platforms on which the solution is deployed.

Qualifications

• Bachelor or above degree in computer science or related majors.
• Familiar with the recent advances in LLM, distributed computing, model deployment;
• At least 3 years or more working experiences;
• Familiar with machine learning algorithms, platforms and frameworks such as PyTorch and Jax.
• Expert in at least one or two programming languages in Linux environment: CUDA, Python, C/C++;
• Bilingual, fluent in Chinese and English is a plus (as candidates need to liaise with Chinese/Mandarin speaking counterparts in China for job purposes)

The following experiences will be a big plus:

• Publication record on top-tier AI/machine learning conferences or journals.
• Experiences in designing and operating large scale systems in cloud computing or machine learning;
• Experiences in CUDA programming and performance tuning (triton).