AI Engineer, LLMs

Logical Intelligence • San Francisco, CA

6h

About The Position

Join our team as an AI Engineer and help us push the boundaries of what's possible in logical reasoning! We
re looking for a motivated individual to design, implement, and refine efficient Large Language Models (LLMs) pipelines for scaled distributed training. You'll be at the forefront of designing and refining algorithms that go beyond the capabilities of traditional LLMs. You'll work closely with a talented team of AI experts, EBM specialists, formal verification engineers, and software developers to create groundbreaking solutions.

Requirements

• Deep understanding of transformers' internals, and ability to make radical changes to the architecture and handle higher-order derivatives
• Expertise in programming languages and tools critical for high-performance computing in Python/C++ and machine learning including Deep Learning frameworks like PyTorch /TensorFlow/JAX
• Expertise in optimizing machine learning systems, including general techniques and LLM-specific optimizations
• Understanding state-of-the-art approaches in LLM reasoning
• Ability to understand complex learning approaches, such as energy-based models
• 3+ years of production experience in ML Infra, DataOps, distributed training. Proficiency with Kubernetes clusters and distributed compute assets
• Strong communication and teamwork skills
• Readiness to explore and promote cutting edge technologies in ML Infrastructure domain and beyond

Nice To Haves

• Demonstrated publications in any of the major conferences
• Experience in EBM or latent reasoning
• Mathematical Reasoning discrete math and logic

Responsibilities

• Implement new reasoning algorithms and models
• Evaluate reasoning approaches, including latent space reasoning
• Pre-train, fine-tune, and modify the State-of-the-Art LLMs
• Optimizing and scaling LLM pipelines
• Adjust frameworks and interfaces to accelerate machine learning development
• Derive practical solutions and integrate them with the results of other teams to provide the best overall resolution

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Education Level: No Education Listed