Junior Machine Learning Engineer

Employment Type: 6-Month Contract / Internship
Location: Zurich, on-site (hybrid)

We are seeking a Junior Machine Learning Engineer / Intern to join our team, assisting in the post-training and optimization of large language models (LLMs), their integration with retrieval-augmented generation (RAG), and multi-agent architectures. The candidate will support improving our LLM-based solutions through applied research, solid implementation, evaluation, and optimizations. You will collaborate closely with experienced machine learning engineers, software engineers, and domain experts to push the frontier of LLM for code generation.

Key Responsibilities:

• Assist in designing LLM training datasets for instruction following, question answering, reasoning, conversational models, and code generation.
• Support the integration of curated content into fine-tuning pipelines.
• Help define instructions for knowledge workers to create LLM training data (code, text, prompts, dialogues) and synthetic data validation processes.
• Support systematic evaluation of LLM performance and benchmarking.
• Participate in refining retrieval-augmented generation (RAG) engines.
• Learn and assist in developing multi-agent architectures and prompt engineering.
• Experiment with new trends and research directions in LLMs under guidance.
• Implement LLM components and integrations in high-quality code, with proper documentation.

Required Qualifications:

• Currently pursuing or recently completed Bachelor’s or Master’s degree in Computer Science, Data Science, or a related field.
• Familiarity with machine learning frameworks (e.g., PyTorch or TensorFlow).
• Basic proficiency in Python.
• Strong communication skills and ability to collaborate effectively.
• Strong analytical and problem-solving abilities.

Preferred Qualifications:

• Basic understanding of LLM architectures like BERT, Llama-3, or similar.
• Familiarity with code generation or low-resource programming languages.
• Interest or experience in reinforcement learning from human feedback (RLHF) or reinforcement learning from verifiable rewards (RLVR).
• Familiarity with data annotation and labeling tools.
• Interest in bias and fairness issues in AI.
• Experience or interest in TypeScript.

 

About DFINITY and the Internet Computer:

DFINITY is a leading contributor to the Internet Computer Protocol (ICP), with a mission to bring the world's compute onto the secure ICP network. Built on its unique third-generation blockchain technology, ICP enables the development and operation of a new generation of unstoppable, tamper-proof, fully decentralized web applications. Its powerful technology can run entire AI models within smart contracts, representing a major advancement for secure AI. Through seamless integration with Bitcoin, Ethereum, and other networks, ICP facilitates multi-chain operations for digital assets and web3.
Join our team of over 250 talented individuals, including world-renowned cryptographers, distributed systems engineers, programming language experts, and industry leaders, who are shaping the future of the internet and web3.   DFINITY was founded in 2016 by entrepreneur and crypto theoretician, Dominic Williams.

All qualified applicants will receive consideration for employment without regard to race, color, religion, gender, gender identity or expression, sexual orientation, national origin, genetics, disability, age, or veteran status.