Edge AI Research Scientist - Equity Options

webAI; Austin (TX); On-site; USD 180,000 - 240,000; Full time

Benefits offered by this job

Competitive salary

Health benefits

401(k) match

Equity options

Health & Wellness stipend

Continuing Education support

Function Health subscription

Free parking for in-office employees

Flexible Time Off

Parental leave

Job summary

webAI in Austin, TX is seeking an AI Research Scientist to join the Intelligence Lab as the core driver of novel model development. You will push boundaries in Transformers, diffusion, and other architectures, collaborating with engineers and leadership to translate breakthroughs into practical, privacy-preserving applications.

We value experimentation, rigorous evaluation, and clear documentation, with a hands-on role that operates in a high-uncertainty environment and a strong emphasis on

Qualifications

• PhD in ML, mathematics, physics, or a highly quantitative field.
• 5 years of post-doc or industrial research with publications or model releases.

Responsibilities

• Design and train novel architectures (Transformers, diffusion, state-space models).
• Lead hypothesis testing, define evaluation benchmarks and loss functions.
• Document findings and mentor engineers on mathematical nuances of models.

Skills

Deep learning

Python

Mathematics

Linear algebra

Probability

Education

PhD in Machine Learning or quantitative field

Tools

PyTorch

TensorFlow

JAX

Job description

webAI in Austin, TX is seeking an AI Research Scientist to join the Intelligence Lab as the core driver of novel model development. You will push boundaries in Transformers, diffusion, and other architectures, collaborating with engineers and leadership to translate breakthroughs into practical, privacy-preserving applications.

We value experimentation, rigorous evaluation, and clear documentation, with a hands-on role that operates in a high-uncertainty environment and a strong emphasis on