Founding Engineer (Systems + ML)

What you will do

• Explore new ML model architectures which give 1000x performance over existing methods for chip design.
• Arcitect and build end-to-end pipelines: high-performance kernels, efficient file IO, training models, latency sensitive inference, CLI/UI layers.
• Talk to customers, deploy in their office, fix prod on the fly.
• Learn quickly.

Requirements

• Expert in PyTorch and CUDA programming.
• Deep understanding of systems - memory, latency, and performance tradeoffs.
• Experience designing, training and deploying custom models (a GPT wrapper does not count).
• Versatility: comfortable toggling between model architecture design, low-level optimization and scaleable system design (N~>10M).
• Proficient with GPU profiling and performance analysis.

Nice to Have

• Exposure to chip design and electronic design automation (EDA).
• Expert in Reinforcement Learning
• Papers in respected conferences and journals.
• Experience with physics-driven ML.

Benefits

Partcl is developing the next generation of chip design automation tools. Using GPUs and AI, we are developing design tools that are 1000x faster than legacy options. That’s the difference between waiting weeks vs. minutes for design feedback.

Schedule

Full-time

Pay

$130K - $300K

Company

San Francisco, CA, US