Intern - Tensor Networks

Zapata AI is the Industrial Generative AI company, revolutionizing how enterprises solve their hardest problems with its powerful suite of Generative AI software. By combining numerical and text-based solutions, Zapata AI empowers industrial-scale commercial, government and military/defense enterprises to leverage large language models and numerical generative models better, faster, and more efficiently delivering solutions to drive growth, savings and unprecedented insight.

With proprietary science and engineering techniques and the Orquestra® platform, Zapata AI is accelerating Generative AI’s impact in Industry.

We are seeking an intern with tensor network experience to complement respective teams within professional services. Projects will require both basic research and software development, with a focus on serving industrial clients. This internship is fully remote (Office hybrid option available, if intern wants)

• Set up and carry out numerical studies to identify and refine new applications of Zapata’s quantum-inspired technologies for generative AI.
• Produce high-quality software implementing the intern’s methods and contribute to Zapata’s in-house software libraries.
• Expand the range of algorithms and models offered withing Zapata’s product, and augment functionalities of existing ones based on feedback from users, customers and researchers.
• Participate in regular group meetings to discuss the intern’s findings and extract high-level insights and future opportunities.

• Completed or enrolled in a STEM degree (MS or PhD).
• Proficiency in working with tensor networks, including tensor decomposition, manipulation, and optimization techniques.
• Familiarity with one or more common tensor network libraries, e.g., ITensors, quimb, tntorch, TeNPy or others.
• Excellent verbal and written communication skills, ideally with prior academic publications.

• Proficiency in Julia and/or Python
• Experience developing software for technical computing.
• Experience in benchmarking and profiling high-performance parallel computing Julia/Python code.
• Experience with GPUs.
• Demonstrated ability to conduct independent research, including literature review, experimental design and data analysis.
• Experience in fast-paced group projects and a willingness to learn new topics in a collaborative setting.

Please upload your CV and a cover letter introducing yourself, explaining why you believe working at Zapata would enhance your career (in 1-2 paragraphs). Cover letter is important for us to understand your candidacy for this role.