Applied AI Engineer

The role

You will develop and optimize a large neural network-based tabular model implemented in Python. You will profile pipelines, rewrite components in Rust or C++, and collaborate with ML researchers to productionize capabilities.

What you will 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 Rust or C++ where Python limits performance
• 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 ML systems handling tabular data

Required qualifications

• 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)

Nice to have

• Experience with tabular ML approaches (transformers, tree/NN hybrids, learned embeddings)
• Working proficiency in C++ and experience bridging Python and C++ (PyBind11, Cython, or custom extensions)
• Familiarity with PyTorch internals or writing custom ops (Rust or C++)
• 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)
• Experience with the Rust async ecosystem (tokio) or SIMD/parallelism crates (rayon, ndarray)

Benefits

• Competitive compensation with salary and equity
• Comprehensive health coverage for you and your dependents
• 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

About the company

Fundamental is an AI research lab pioneering the future of enterprise decision-making. Their flagship model, NEXUS, is a Large Tabular Model purpose-built for structured records.