GPU Engineer

About Us

Axiom is using zero knowledge proofs to scale smart contract applications in a new way. Our recent V2 release on mainnet allows smart contracts to compute over the entire history of Ethereum in a fully trustless way. This is critical infrastructure to help blockchain applications scale and will empower developers to build novel decentralized applications with the power of ZK.

We are a small but incredibly strong team backed by the best investors in crypto. We’re looking for exceptional people who are excited to work on challenging and impactful problems in a fast-paced environment. Our office is in NYC and working in person in NY is our strong preference, but we're open to remote candidates as well.

The Role

We are looking for a GPU Engineer who will implement and optimize cutting-edge cryptographic code for zero-knowledge proof generation on GPUs. You will be responsible for implementing ZK provers in CUDA, investigating GPU performance of new proof systems, and optimizing the end-to-end performance of the overall system. Your work will require a strong eye for performance and the ability to strike a balance between performance and flexibility.

No prior experience with zero-knowledge proofs is required, but the ability to rapidly ramp up on this new space is critical.

Who you are

• You have experience writing performant code in CUDA, including handling memory pipelining to deliver an optimized end-to-end solution.

• You have a strong background in math/cryptography, or strong problem-solving skills and interest in zero-knowledge proofs.

• You are proficient in Rust, C/C++, or a similarly performant backend language. We work primarily in Rust.

• You are extremely detail oriented and have a security mindset.

• You’ve worked on challenging software projects and have an eye for performance.

• You’re a clear communicator and value collaboration.

• You have a strong bias towards action.

• You have an extremely strong work ethic.

Bonuses

• You have hands-on experience working with computer architecture, e.g. compilers, ISAs, processors.

• You have experience with compiler frameworks like LLVM or MLIR.

• You've worked with aspects of low level computer architecture like RISC-V or programmed FGPAs or other custom hardware.

• You've done research in mathematics, theoretical computer science, or theoretical physics.

We welcome diverse backgrounds. Some non-traditional profiles that may be a good fit include high-frequency trading, math/physics/CS academia, and hardware engineering.

Why join us

• You’ll be one of the earliest employees with ownership to make key technical/product decisions.

• You’ll be working on groundbreaking tech that has the potential to be a key piece of infrastructure for the crypto industry.

• You’ll receive a competitive salary with equity.

• You will work with and receive mentorship from an exceptional team.

Our Team

Our founders Yi Sun and Jonathan Wang met while studying math at Harvard and have been working on ZK since early 2022. Together, they wrote some of the most performant ZK circuits for elliptic curve cryptography and Ethereum data reads before starting Axiom.

Before Axiom, Yi researched probability and machine learning as a professor at UChicago Statistics and did a math PhD at MIT. He was an advisor for Gauntlet and Scroll and worked at an HFT startup from inception to profitable trading. Jonathan published number theory research in top math journals and did a math PhD at UChicago.

The rest of our small but incredibly strong engineering team includes alums of Applied Intuition, Coinbase, Color, Snap, Neon, and Google. We look forward to working with you!