AI Research Engineer, Code Metal

• Ph.D. in Computer Science or a closely related field
• Prior LLM research experience
• Comfortable programming in Python and familiar with frameworks such as PyTorch and HuggingFace

Desirable

• Our mission is to enable hardware deployment at the speed of software development. We are working towards automatic code transpilation and optimization for any hardware application
• In this role, you will collaborate with a small team of talented researchers on ambitious, greenfield projects in generative AI and reinforcement learning
• Design, execute, and analyze experiments with a high degree of independence
• Contribute to core models and frameworks
• Create high-quality datasets (both in-the-wild and synthetic)
• Perform literature reviews and implement new techniques from papers
• Contribute to publications, present at conferences and workshops, etc
• Research Areas of Interest:
• An incomplete list of current and near-term research directions:
• Contrastive representation learning
• Steerability and guided decoding
• Tractable probability models
• Code-specific architectures
• LLM fine-tuning, post-training, RLHF