Fundamental is seeking an Applied AI Engineer to join our mission-driven team. You will take part in the development and optimization of a large neural network-based tabular model, directly impacting the performance and scalability of our core systems.
What You'll Do
- Take part in development and optimization of a large neural network-based tabular model implemented in Python.
- Profile training and inference pipelines to identify performance bottlenecks.
- Rewrite critical components in C++ (via PyBind11 or custom extensions) where Python limits us.
- Improve memory efficiency, latency, and throughput across model pipelines.
- Ensure correctness, numerical stability, and reproducibility as the model evolves.
- Collaborate with ML researchers on productionizing new capabilities.
- Maintain clean abstractions, comprehensive tests, and clear documentation.
- Shape architectural decisions for our ML systems handling tabular data.
What We're Looking For
- Strong software engineering fundamentals with expert-level Python and C++.
- Hands-on experience bridging Python and C++ (PyBind11, Cython, 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).
Nice to Have
- Experience with tabular ML approaches (transformers, tree/NN hybrids, learned embeddings).
- Familiarity with PyTorch internals or writing custom ops.
- Experience optimizing training loops, data pipelines, or inference engines.
- Background in numerical computing or systems programming.
- Exposure to large-scale ML infrastructure (distributed training, batching, caching).
Technical Stack
- Python, C++, PyBind11, PyTorch
Benefits & Compensation
- Competitive compensation with salary and equity.
- Comprehensive health coverage, including medical, dental, vision, and 401K.
- Fertility support.
- Paid parental leave for all new parents, inclusive of adoptive and surrogate journeys.
- Relocation support for employees moving to join the team in one of our office locations.
Fundamental is an equal opportunity employer committed to a low-ego culture that values diversity of thought, ownership, and bias toward action.
