AI Research Engineer (Pre-training - LLM & Multi-Modal)

Responsibilities

• Large-Scale Pre-Training
• Conduct foundational pre-training for LLMs and Multi-Modal models (integrating text, vision, audio, or other modalities) on large, distributed servers equipped with multi-nodes & thousands of NVIDIA GPUs.
• Architecture & Alignment Innovation
• Design, prototype, and scale innovative architectures, tokenizers, and cross-modal alignment layers to enhance model intelligence and multi-modal understanding.
• Data Strategy: Source, filter, and curate massive-scale textual and multi-modal datasets, establishing robust data pipelines for efficient pre-training.
• Experimental
• Research: Independently and collaboratively execute experiments, analyze results, and refine training methodologies for optimal performance and token efficiency.
• Optimization & Debugging: Investigate, debug, and eliminate bottlenecks in model efficiency, computational performance, and multi-modal alignment stability during long training runs.
• System Scalability
• Contribute to the advancement of distributed training systems to ensure seamless scalability and hardware efficiency on target platforms.

Languages