Senior Specialist AI Engineer - Foundations

Company Description

Linked In is the worlds largest professional network, built to create economic opportunity for every member of the global workforce. Our products help people make powerful connections, discover exciting opportunities, build necessary skills, and gain valuable insights every day. Were also committed to providing transformational opportunities for our own employees by investing in their growth. We aspire to create a culture thats built on trust, care, inclusion, and fun where everyone can succeed.

Job Description

This role will be based in Sunnyvale, San Francisco, or New York City. At Linked In, our approach to flexible work is centered on trust and optimized for culture, connection, clarity, and the evolving needs of our business. The work location of this role is hybrid, meaning it will be performed both from home and from a Linked In office on select days, as determined by the business needs of the team.

Job Description

As a Senior AI Scientist, you will drive Linked In's efforts in cutting-edge AI research and development, focusing on large-scale foundation models. In this role, you will specialize in developing, training, and optimizing foundation models, including Large Language Models (LLMs), utilizing state-of-the-art deep learning architectures such as multi-billion parameter transformers and scaling them to serve Linked In's global user base. Your work will emphasize the design and deployment of high-performance AI systems, leveraging GPUs and other hardware accelerators for large-scale training and inference.

You will collaborate with a world-class team of researchers and engineers to develop and advance the next generation of AI models to push the boundaries of AI at scale, working on complex problems such as distributed training, model parallelism, and system co-design. A successful candidate will be comfortable operating at the intersection of AI research and engineering, with a deep understanding of the latest advancements including but not limited to large model training, post-training techniques for planning and reasoning, fine-tuning strategies, reinforcement learning techniques.

Responsibilities

* Lead the research, design, and development of large-scale foundation AI models, with a focus on LLMs, transformer architectures.

* Publish the findings and innovations as the latest academic and industry research in top notch core AI venues such as ICML, NeurIPS, ICLR, ACL.

* Drive innovation in large-scale model training, leveraging GPUs and specialized hardware for optimal performance.

* Develop and maintain scalable AI pipelines, optimizing training and inference workflows to handle vast amounts of data.

* Collaborate with cross-functional teams of machine learning engineers, infrastructure engineers, and AI researchers to deliver impactful AI innovations across Linked In's products.

Qualifications

Basic Qualifications

* BA/BS Degree in Computer Science, Machine Learning or related technical discipline, or related practical experience.

* 2+ years experience in machine learning or AI engineering.

* 2+ years experience in design and development of algorithmic solutions

* 2+ years experience with programming languages such as Java, Python, etc.

Preferred Qualifications

* 4+ years of hands-on experience in large-scale model training, model-system co-design, or large AI systems.

* PhD in Computer Science, Machine Learning, or related technical discipline with a focus on advanced AI/ML techniques

* Proven experience in building and optimizing large-scale AI/ML models using frameworks such as PyTorch, PySpark, and CUDA.

* Several recent publications in top AI conferences on relevant areas including but not limited to Large Scale Model Training and LLM Algorithm Optimization, deep learning, reinforcement learning, foundation models etc.

* Expertise in distributed training, model parallelism, and hardware acceleration for AI workloads, including GPUs and TPUs.

* Demonstrated ability to solve complex AI challenges in large-scale environments, optimizing models for efficiency and scalability.

* Strong proficiency in AI system design, model optimization, and the development of production-quality AI pipelines.

* Published work in AI research or a significant contribution to industry-leading AI technologies.

Suggested Skills

* Large Language Models (LLM)

* Generative AI (GAI)

* AI Research and Development

* GPU and Hardware Acceleration

* Distributed AI Systems and Model Parallelism

You will benefit from our culture:
We strongly believe in the well-being of our employees and their families. That is why we offer generous health and wellness programs and time away for employees of all levels. Linked In is committed to fair and equitable compensation practices. The pay range for this role is $139,000 - $229,000. Actual compensation packages are based on several factors that are unique to each candidate, including but not limited to skill set, depth of experience, certifications, and specific work location.

This may be…