Note: The job is a remote job and is open to candidates in USA. The company is a seed-stage startup in the nuclear energy sector partnering with power plants to develop AI models for nuclear data and content. The Machine Learning Engineer will build and optimize AI systems for document understanding, retrieval-augmented generation, domain-adapted language models, and nuclear information analysis while collaborating with scientists and nuclear experts.
Responsibilities
- Develop and implement AI models to understand complex nuclear content and concepts
- Design and optimize AI models for efficient data navigation and document indexing
- Collaborate with scientists to leverage supercomputing resources for model training
- Implement safeguards to prevent AI hallucinations and biases during search and data analysis
- Create robust, scalable, and maintainable code for our AI platform
- Contribute to the development of open-source AI tools for the nuclear sector
- Work closely with nuclear experts to ensure AI models accurately interpret industry-specific content
Skills
- MUST BE A U.S. Citizen
- 3+ years of experience in AI/ML software engineering
- Strong background in natural language processing, machine learning, and large language and information retrieval models
- Proficiency in Python and experience with PyTorch
- Strong understanding of AI models, particularly in document understanding and information retrieval
- Hands-on experience implementing and fine-tuning Retrieval-Augmented Generation (RAG) systems for domain-specific applications
- Experience with the HuggingFace ecosystem
- Experience with large-scale data processing and distributed computing
- Familiarity with open-source development practices and version control systems
- Excellent problem-solving skills and ability to work in a fast-paced, innovative environment
- Strong communication skills to collaborate with cross-functional teams and external partners
- Commitment to safety and security in dealing with nuclear-related information
- Experience with high-performance distributed computing environments is a plus
- Familiarity with or interest in the nuclear energy sector
Benefits
- Fully remote role
Company Overview