About the Role
We’re hiring an AI Research Engineer to push the frontier of agentic LLMs and reinforcement learning for our agentic code generation tool. You’ll design and run experiments, build agents, curate datasets from complex technical documents, and create rigorous evaluations. You’ll write production‑quality research code and work closely with engineering to ship improvements to customers. Impact through research and building is key.
Responsibilities
- Design and implement multi‑agent and RL approaches for agentic code generation and tool‑use.
- Build research prototypes that integrate with our agentic code generation tool; collaborate to productionize wins.
- Create evaluation suites: task specs, pass/fail checkers, coverage, cost/latency dashboards.
- Acquire and curate datasets from PDFs/logs/tables; generate synthetic data where appropriate; maintain data cards and licensing.
- Analyze experiments with disciplined ablations; document results and decisions.
- Stay current on LLM agents, RL (offline/online, RLHF/RLAIF), constrained decoding, and program synthesis.
Requirements
- PhD in CS/AI/ML (or equivalent research experience) with publications ideally in multi‑agent RL, agentic AI, or RL for language/code.
- Strong Python and ML framework experience (PyTorch preferred; JAX/HF a plus).
- Demonstrated ability to turn research into working systems; reproducibility mindset (tests, seeds, configs, logging).
- Experience designing eval harnesses and success metrics for sequential/agentic tasks.
- Comfortable with data acquisition/curation from documents/logs; good instincts about data quality and licenses.
- Clear communicator who partners well with engineers.
Skills
- Python
- PyTorch
- JAX
- Hugging Face
- LLM agents
- Reinforcement Learning (RL)
- Offline RL
- Online RL
- RLHF
- RLAIF
- Constrained decoding
- Program synthesis
- Data acquisition
- Data curation
- Evaluation harnesses
- Success metrics
- Sequential tasks
- Agentic tasks
- Semiconductor/chip domains
- Technical specifications
Location
- Remote
Work Type
- Full-time
Experience Level
- PhD or equivalent research experience
Education Level
- PhD in CS/AI/ML
About the Company
- Normal Computing is an Equal Opportunity Employer. We celebrate diversity and are committed to creating an inclusive environment for all employees.
- Normal Computing is committed to providing reasonable accommodations to individuals with disabilities.
- By submitting your application, you agree that Normal Computing may collect, use, and store your personal information for employment-related purposes in accordance with our Privacy Policy.
Equal Opportunity
- All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, veteran status, or any other legally protected status.
