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.