Member of Technical Staff – Foundation Model Architecture, AI Infrastructure
United States
Full Time
20 hours ago
$100,000 - $220,000 USD
No H1B
Key skills
PyTorchML
About this role
Role Overview
- Define and scale the core operator intelligence layer
- Design and refine transformer variants for structured spatial domains
- Expand distributed training beyond 45TB-scale datasets
- Improve generalization across heterogeneous operator distributions
- Develop graph-transformer systems for multi-entity interactions
- Ship expanded operator capabilities into production
Requirements
- Deep experience in foundation model architecture
- Transformer variants (sparse, hierarchical, graph-based)
- Distributed training systems
- Production ML system design
- Scaling structured datasets
- Writing clean, maintainable, high-quality code
- Strong software engineering fundamentals
- Experience with modern ML stacks (e.g., PyTorch and distributed training ecosystems)
- Strong CI, regression testing, and validation discipline
Tech Stack
- PyTorch
Benefits
- High ownership at Series A stage
- Opportunity to define a foundational abstraction layer early