Applied AI Engineer

📋 Description

• Develop a large neural network tabular model in Python
• Profile training and inference pipelines to identify bottlenecks
• Rewrite core components in Rust (via PyO3) or C++ (via PyBind11)
• Improve memory, latency, and throughput across model pipelines
• Ensure numerical stability and reproducibility as the model evolves
• Collaborate with ML researchers to productionize new capabilities

🎯 Requirements

• Strong software engineering fundamentals with expert-level Python and Rust
• Hands-on experience bridging Python and Rust (PyO3, maturin, or custom extensions)
• Experience developing and maintaining ML models in production
• Strong understanding of neural networks
• Track record of optimizing performance-critical code
• Strong profiling and debugging skills (CPU, memory, latency)

🎁 Benefits

• Competitive compensation with salary and equity
• Comprehensive health coverage for you and your dependents
• Paid parental leave for all new parents, including adoptive/surrogate journeys
• Relocation support for employees joining the team at our office locations
• A mission-driven, low-ego culture valuing diversity and action