# Senior ML/AI Modeler, Risk Automation Machine Learning

**Company:** [Square](https://hotfix.jobs/companies/square)
**Location:** Remote
**Role:** ML Engineering
**Salary:** $195k – $343k/yr
**Experience:** 8+ years
**Skills:** Machine Learning, Artificial Intelligence, LLMs, Natural Language Processing, Python, PyTorch, Snowflake, Apache Airflow, GCP, AWS, Transformer Models, Prompt Engineering, MLOps, Claude Code, Codex
**Posted:** 2026-08-19

> Leads the design, deployment, and optimization of agentic and generative AI systems that automate risk and compliance investigations at scale. Requires 8+ years of machine learning modeling experience, production ML expertise, and advanced technical education.

## Job Description

## Responsibilities
- Experiment with and deploy AI copilot and autopilot systems at scale to improve analyst productivity and eliminate manual decision loops.
- Own end-to-end systems, including API integrations with disparate data sources, prompt tuning, orchestration, metrics, evaluation, productionization, and monitoring.
- Use payment transactions, connected-user and asset graphs, unstructured text, and user profiles to build transformer-based ML models for downstream detection tasks.
- Collaborate with product, platform, engineering, and operations stakeholders to deploy production-grade systems and tune ongoing performance.
- Apply Python ML tools, LLMs, PyTorch, Snowflake, Airflow-based tools, data platforms, and GCP and AWS cloud services.
- Use agentic tools such as Claude Code, Codex, and OpenClaw for research, development, DevOps, and documentation.
- Own and drive strategic roadmaps and priorities across cross-functional stakeholders.

## Requirements
- 8+ years of machine learning modeling experience.
- Full-stack machine learning experience preferred.
- Master's or advanced degree in computer science, data science, operations research, applied mathematics, statistics, physics, or a related technical field.
- 3+ years of AI engineering and large language model experience preferred.
- Background in traditional natural language processing techniques preferred.
- End-to-end experience building and deploying performant ML/AI systems in batch and real-time production environments.
- Experience independently owning and driving high-impact programs with multiple cross-functional stakeholders.
- Curious, growth-oriented mindset and ability to apply first-principles thinking.

## Compensation and Benefits
- ATS salary range: $194,500–$343,100 USD.
- Pay varies by location, skills, experience, qualifications, and market conditions; U.S. locations are categorized into compensation zones.
- Benefits include remote work, medical insurance, flexible time off, retirement savings plans, and modern family planning.

## Similar jobs

- [Senior Software Engineer, ML Infrastructure Platform](https://hotfix.jobs/jobs/42001ce3-19de-49b5-8883-d001ca38069f) - Nuro - Mountain View, CA - $194k – $291k/yr
- [Senior Software Engineer, Autonomy Capabilities](https://hotfix.jobs/jobs/ae9ec2ef-285e-42da-9740-0b3ce5faf8cf) - Shield AI - San Mateo, CA - $196k – $294k/yr
- [Senior Machine Learning Engineer, Payments](https://hotfix.jobs/jobs/4dcd0258-2a7a-4e22-a27e-f2e1426de86e) - Airbnb - Remote - $191k – $223k/yr
- [Senior Machine Learning Infrastructure Engineer, Embedding Platform](https://hotfix.jobs/jobs/f7e37122-fd13-4390-b7b8-27c86092f1ae) - Reddit - Remote - $191k – $267k/yr
- [Senior Applied AI Engineer](https://hotfix.jobs/jobs/03330d35-9411-4346-b3db-57b680d941c7) - Mintlify - San Francisco, CA - $190k – $265k/yr

**Apply:** https://hotfix.jobs/jobs/303b4f2e-9bc8-4034-8fa1-d17c71bf6f1e
**Canonical:** https://hotfix.jobs/jobs/303b4f2e-9bc8-4034-8fa1-d17c71bf6f1e