Apply on Employer Site

Forterra · 1 day ago

Senior AI Engineer - Applied Scientist

East Palo Alto, CA

Full-time

Onsite

Senior Level, Lead/Staff

$145K/yr - $185K/yr

3+ years exp

Forterra is a leading provider of autonomous systems for ground-based movement in the working world. They are seeking a Senior AI Engineer with deep expertise in generative modeling and multimodal AI systems to design and implement advanced models for understanding complex agent interactions in dynamic environments.

Defense & Space

Growth Opportunities

No H1B

Security Clearance Required

U.S. Citizen Only

Responsibilities

Develop neural network architectures for modeling sequential agent interactions using generative modeling approaches

Implement Transformer-based models that encode agent and contextual features with attention mechanisms for capturing interactions

Research and integrate goal-conditioned models that leverage structured representations before generating complete sequences

Prototype and optimize learning models into lightweight modules suitable for real-time inference

Collaborate with cross-functional teams to ensure seamless integration of models into the broader AI stack

Benchmark and visualize model performance using real-world and simulated datasets

Optimize models for low-latency inference and interpretability of output distributions

Qualification

Generative modelingTransformersProbabilistic modelsPyTorch/TensorFlowMultimodal sensor processingSequential modelingDeep learningSoftware engineeringResearchCollaboration

Required

Deep understanding of probabilistic generative models for sequential forecasting tasks

Strong experience with Transformers and graph-based modeling for complex relational data

Familiarity with multimodal sensor processing and representation learning

Proven ability to work with uncertainty quantification and latent variable inference

Experience implementing models in PyTorch/TensorFlow and deploying them in production environments

M.S. or Ph.D. in Computer Science, Machine Learning, Robotics, or related field

3+ years of experience with sequential modeling, generative architectures, or forecasting systems

Strong foundation in applied deep learning, especially variational inference and attention mechanisms

Solid software engineering background with emphasis on clean, modular, and optimized implementations

Preferred

Publications or project experience in forecasting, agent modeling, or scene understanding

Familiarity with graph-based or vectorized representations of structured environments

Experience with diffusion model optimization or goal-conditioned decoding

Contributions to open-source frameworks related to generative modeling or sequential forecasting

Interest in bridging high-fidelity modeling with real-time applications

Benefits

Premium Healthcare Benefits: Three plan options, including an HSA-eligible plan, with Forterra covering 80% of the plan premium for you and your dependents.

Basic Life/AD&D, short and long-term disability insurance plans 100% covered by Forterra, plus the option to purchase additional life insurance for you and your dependents.

Extremely generous company holiday calendar including a winter break in December.

Competitive paid time off (PTO) offering 20 days accrued per year.

A minimum of 7 weeks fully paid parental leave for birth/adoption.

$9k annual tuition reimbursement or professional development stipend.

Fully stocked beverage refrigerators with all the Celsius your little heart desires.

401(k) retirement savings plan, including traditional, Roth 401(k), and after-tax deferral with company match up to 4%.