Latest AI JobsBrowse Jobs
Attentive logo

Staff Software Engineer, Machine Learning

AttentiveNew York, NYEngineering
Apply on company site
Salary
$320K - $360K
Job type
Full-time
Experience
Senior · 6+ yrs
Posted
21 Sept 2026
Status
Open · checked 7d ago

About this role

Building, scaling, and operating production-grade machine learning systems that drive real-time personalization forms the core of daily operations on this team. Engineers partner closely with product and engineering departments to power tailored experiences across an AI marketing platform utilized by thousands of brands.

Operating with a high degree of ownership, this team focuses on raising technical standards while automating data analysis, model development, validation, and implementation from modern research. Ideal candidates possess a background in constructing scalable systems and data-driven products alongside a strong track record of leading cross-functional machine learning initiatives.

What you will do

  • Design and operate production-grade machine learning systems that power real-time personalization across the platform.
  • Partner closely with product and engineering teams to deliver scalable data-driven products.
  • Protect system quality and prevent regressions through rigorous testing techniques and direct code contributions.
  • Drive long-term architectural improvements and advocate for larger-scale technical changes.
  • Build automated processes for large-scale data analyses, model development, model validation, and model implementation.
  • Lead complex, cross-functional machine learning initiatives across different teams.

What you need

  • Bring professional experience building software systems for at least 6 years, with ideal total experience around 10 years.
  • Demonstrate hands-on proficiency with Python and tools such as TensorFlow, PyTorch, XGBoost, Pandas, Matplotlib, SQL, or Spark.
  • Possess extensive experience applying machine learning and data analysis to construct scalable systems.
  • Show a proven track record of creating efficient, automated workflows for model development, validation, and implementation.
  • Demonstrate leadership experience managing cross-functional machine learning projects across multiple teams.

Benefits

  • Health and wellness perks
  • Equity compensation
  • Competitive base salary ranging from $320,000 to $360,000 annually

Skills and tech stack

Questions candidates ask

What is the salary range for this position?

The US base salary range for this full-time position is $320,000 - $360,000 annually. Compensation also includes equity and benefits, with salary ranges determined by role, level and location.

What technologies are used in the infrastructure and backend?

The infrastructure runs primarily in Kubernetes hosted in AWS’s EKS alongside tooling like Istio, Datadog, Terraform, CloudFlare, and Helm. The backend consists of Java and Spring Boot microservices built with Gradle, coupled with DynamoDB, Kinesis, AirFlow, Postgres, Planetscale, and Redis.

What machine learning tools and frameworks does the team utilize?

Automation is driven by custom and open source machine learning models built with Python, Metaflow, HuggingFace, PyTorch, TensorFlow, and Pandas.

What should I do if I do not meet every single requirement listed?

If you do not meet all the requirements listed here, the company still encourages you to apply. No job description is perfect, and there may be another opportunity that closely matches your skills and experience.

Ready to apply?

The application is on the official Attentive careers page.

Apply on company site