> Markdown version of [/jobs/ext/2303250-ai-engineer-reinforcement-learning](https://www.wearedevelopers.com/jobs/ext/2303250-ai-engineer-reinforcement-learning). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md). --- # AI Engineer - Reinforcement Learning - **Company:** LOGICAL LOGICAL - **Location:** Alameda, CA, United States - **Experience:** Experienced - **Contract:** Permanent contract - **Skills:** Artificial Intelligence, C++ (Programming Language), Distributed Computing Environment, Formal Verification, Python (Programming Language), Machine Learning, Performance Tuning, Tensorflow, DataOps, Software Engineering, Reinforcement Learning, Rust (Programming Language), High Performance Computing, Pytorch, Large Language Models, Deep Learning, Kubernetes, Optimization Algorithms, Machine Learning Operations, Golang, Programming Languages - **Published:** August 30, 2026 - **Apply:** https://www.disabledperson.com/jobs/74655941-ai-engineer-reinforcement-learning ## About the Role * 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 Bonus Points: * Demonstrated publications in any of the major conferences * Experience in EBM or latent reasoning * Demonstrated publications in any of the major conferences * Mathematical Reasoning - discrete math and logic ## Description At Logical Intelligence, we're revolutionizing software development with AI-powered formal verification. We've developed groundbreaking agents that provide mathematical guarantees of code correctness, ensuring that software behaves exactly as intended while proactively identifying bugs and security vulnerabilities. Our novel foundation model enables scalable, precise reasoning for formally verifiable code across Rust, Golang, and smart contract VMs. We've won a well-known formal verification benchmark called PutnamBench, which consists of 672 hard math problems from the William Lowell Putnam Exam, the oldest collegiate mathematics competition in North America. Backed by a world-class team - including ICPC champions, a Fields Medalist and an ACM Turing Award winner - we're building the future where all code is provably correct. About the role 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. What you'll do * 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 ## Related Videos - [How AI Models Get Smarter](https://www.wearedevelopers.com/videos/1374-how-ai-models-get-smarter) - [Photonic Computing: Programming a New Class of AI Accelerators (incl. Live Coding)](https://www.wearedevelopers.com/videos/100196-photonic-computing-programming-a-new-class-of-ai-accelerators-incl-live-coding) - [Go with the Flow: Stop the Leaks Before Your Memory's a Waterfall!](https://www.wearedevelopers.com/videos/100073-go-with-the-flow-stop-the-leaks-before-your-memory-s-a-waterfall) - [Serverless deployment of (large) NLP models ](https://www.wearedevelopers.com/videos/158-serverless-deployment-of-large-nlp-models) - [Scoring 2000 Products per Request: Performance Pitfalls in Golang](https://www.wearedevelopers.com/videos/2073-scoring-2000-products-per-request-performance-pitfalls-in-golang) - [The LLM Evolution: From Sequence Imitation to Verifiable Reasoning](https://www.wearedevelopers.com/videos/100343-the-llm-evolution-from-sequence-imitation-to-verifiable-reasoning) ## Related Articles - [What Are Large Language Models?](https://www.wearedevelopers.com/magazine/304-what-are-large-language-models) - [How to Become an AI Engineer](https://www.wearedevelopers.com/magazine/331-how-to-become-an-ai-engineer) - [MLOps And AI Driven Development](https://www.wearedevelopers.com/magazine/82-mlops-and-ai-driven-development) - [MLOps – What’s the deal behind it?](https://www.wearedevelopers.com/magazine/125-mlops-what-s-the-deal-behind-it) - [Graph and AI Trends 2026: Why Is AI Running but Not Yet Delivering?](https://www.wearedevelopers.com/magazine/680-graph-and-ai-trends-2026-why-is-ai-running-but-not-yet-delivering) - [Got AI ideas but no money? Here are 10 free ways to level up your AI skills with Google Cloud](https://www.wearedevelopers.com/magazine/600-got-ai-ideas-but-no-money-here-are-10-free-ways-to-level-up-your-ai-skills-with-google-cloud)