Gpu Engineer

Summary

Join Axiom, a company utilizing zero-knowledge proofs to scale smart contract applications, as a GPU Engineer. You will implement and optimize cryptographic code for zero-knowledge proof generation on GPUs, focusing on CUDA implementation and performance optimization. Prior ZK proof experience isn't required, but rapid learning is crucial. The role demands proficiency in CUDA, a strong math/cryptography background or problem-solving skills, and expertise in Rust, C/C++, or a similar language. Axiom offers a competitive salary with equity, the opportunity to work on groundbreaking technology, and mentorship from an exceptional team. While the office is in NYC, remote work is considered.

Requirements

• Experience writing performant code in CUDA, including handling memory pipelining to deliver an optimized end-to-end solution
• Strong background in math/cryptography, or strong problem-solving skills and interest in zero-knowledge proofs
• Proficiency in Rust, C/C++, or a similarly performant backend language
• Extremely detail oriented and have a security mindset
• Worked on challenging software projects and have an eye for performance
• Clear communicator and value collaboration
• Strong bias towards action
• Extremely strong work ethic

Responsibilities

• Implement ZK provers in CUDA
• Investigate GPU performance of new proof systems
• Optimize the end-to-end performance of the overall system

Preferred Qualifications

• Hands-on experience working with computer architecture, e.g. compilers, ISAs, processors
• Experience with compiler frameworks like LLVM or MLIR
• Worked with aspects of low level computer architecture like RISC-V or programmed FGPAs or other custom hardware
• Done research in mathematics, theoretical computer science, or theoretical physics

Benefits

• Competitive salary with equity
• Mentorship from an exceptional team