# Applied AI Engineer

**Company:** [Fundamental Research Labs](https://hotfix.jobs/companies/fundamental-research-labs)
**Location:** Remote
**Role:** ML Engineering
**Skills:** Python, Rust, C++, Pyo3, Pybind11, PyTorch, Neural Networks, Machine Learning, Performance Optimization, Profiling, Maturin, Cython
**Posted:** 2026-05-20

> Develop and optimize large neural network-based tabular models. Profile and rewrite performance-critical components in Rust and C++ to improve efficiency, latency, and throughput for enterprise AI systems.

## Job Description

## Key Responsibilities
- 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 (via PyO3 or custom extensions) where Python limits us, with C++ (via PyBind11 or custom extensions) as a secondary option where appropriate
- 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

## Must Have
- Strong software engineering fundamentals with expert-level Python and Rust
- Hands-on experience bridging Python and Rust (PyO3, maturin, or custom extensions)
- Working proficiency in C++ and 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 (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, including medical, dental, vision, and 401K
- 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
- A mission-driven, low-ego culture that values diversity of thought, ownership, and bias toward action

## Similar roles

- [CoDesign & NextGen Performance Engineer](https://hotfix.jobs/jobs/2cb728b7-a9ea-44b0-b16a-47364021f7be) - Cerebras Systems - Sunnyvale, CA
- [Research Engineer, Privacy](https://hotfix.jobs/jobs/3220c5d0-0e5f-4869-b133-1ed126b4208a) - OpenAI - San Francisco, CA - $380k – $445k/yr
- [Research Engineer](https://hotfix.jobs/jobs/b4d18018-e64d-4828-a8b4-303225328e17) - Console - San Francisco, CA - $200k – $350k/yr
- [Software Engineer, AI Platform](https://hotfix.jobs/jobs/7ddad000-3e1c-47cb-a45b-31d8e72abc4f) - Notion - San Francisco, CA - $180k – $201k/yr
- [Machine Learning Engineer](https://hotfix.jobs/jobs/97713038-58cb-46c5-a9c7-0e1554e1ba4f) - Liftoff - Remote - $215k – $275k/yr

**Apply:** https://hotfix.jobs/jobs/72266b16-d222-4a0a-87a1-060a32e322a9
**Canonical:** https://hotfix.jobs/jobs/72266b16-d222-4a0a-87a1-060a32e322a9