Reinforcement Learning AI Engineer for Verified Code

Logical Intelligence; San Francisco (CA); On-site; USD 180,000 - 240,000; Full time

Job summary

Logical Intelligence is seeking an AI Engineer to design, implement, and refine efficient LLM pipelines for scalable distributed training. You will join a team of AI experts and formal verification engineers to push the boundaries of logical reasoning and code correctness.

You will work on pre-training, fine-tuning, and optimizing state-of-the-art LLMs, integrating results with other teams to deliver robust, verifiable AI solutions across Rust, Go, and smart contract tooling.

Qualifications

• 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
• Experience with basic distributed optimization techniques
• Familiarity with torch.compile or similar performance optimization tools
• Understanding of LLM architectures and LLM fine tuning internals
• 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

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

Skills

Transformers internals

Python programming

C++ programming

Distributed training

LLM optimization

Energy-based models

ML infra

Communication skills

Team collaboration

Tools

PyTorch

TensorFlow

JAX

Kubernetes

torch.compile

Job description

Logical Intelligence is seeking an AI Engineer to design, implement, and refine efficient LLM pipelines for scalable distributed training. You will join a team of AI experts and formal verification engineers to push the boundaries of logical reasoning and code correctness.

You will work on pre-training, fine-tuning, and optimizing state-of-the-art LLMs, integrating results with other teams to deliver robust, verifiable AI solutions across Rust, Go, and smart contract tooling.