AI Engineer - Reinforcement Learning
Logical Intelligence· AI RESEARCH
Hybrid (San Francisco)full-timePosted 6mo ago
About the role
AI Engineer role at Logical Intelligence focused on designing, implementing, and refining efficient LLM pipelines for scaled distributed training, with emphasis on advanced reasoning algorithms beyond traditional LLMs in a formal verification context.
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
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
Skills
Required:Transformers internalsPython/C++PyTorch / TensorFlow / JAXLLM optimizationDistributed training / ML InfraKubernetesCommunication & teamwork
Nice to have:Energy-based modelsEbm / latent reasoning
Benefits
- Note: Benefits not explicitly listed in the job description.
Last updated Aug 20, 2026. Interested? Apply on the company site.
About Logical Intelligence
AI Certainty for Critical Systems
Logical Intelligence engineers deterministic AI for critical systems. We deliver mathematically verified intelligence beyond LLMs for autonomy, finance, & defense. Explore Math AI & EBMs.
AIdeterministic AIcritical systemsautonomyfinancedefenseMath AIEBMs
More open roles at Logical Intelligence
- AI Researcher - Frontiers & Reasoning7mo agoSan Francisco · AI RESEARCH
- AI Engineer in Synthetic Data7mo agoHybrid (San Francisco) · AI RESEARCH