Sr. ML Engineer
Visa Austin, USASr. ML Engineer
About Us
Visa is a world leader in payments technology, facilitating transactions between consumers, merchants, financial institutions and government entities across more than 200 countries and territories, dedicated to uplifting everyone, everywhere by being the best way to pay and be paid.
At Visa, you'll have the opportunity to create impact at scale - tackling meaningful challenges, growing your skills and seeing your contributions impact lives around the world.
Join Visa and do work that matters - to you, to your community, and to the world. Progress starts with you.
Job Description
As the world's leader in digital payments technology, Visa's mission is to connect the world through the , convenient, reliable, and secure payments network, enabling individuals, businesses, and economies to thrive. Our advanced global processing network, VisaNet, provides secure and reliable payments around the world, and is capable of handling more than 65,000 transaction messages a second.
The company's dedication to innovation drives the rapid growth of connected commerce on any device and fuels the dream of a cashless future for everyone, everywhere. As the world moves from analog to digital, Visa is applying our brand, products, people, network, and scale to reshape the future of commerce.
Visa AI Studio is Visa's AI operating system: a single platform for building, deploying, and operating predictive models, foundation models, and AI agents at global scale. It gives every team at Visa a common, self service way to train and experiment, build with generative AI, develop and run production AI agents, manage features and memory, and operate everything with governance and observability built in. Visa AI Studio is the foundation for how AI gets built across the company.
We are looking for an Machine Learning Engineer to join the AI Engineering Platform team within Visa AI Studio, working on the systems that let every team at Visa train, deploy, and operate models and agents themselves.
This is a modern AI engineering role: it sits deliberately at the intersection of core AI/ML knowledge and systems ands software engineering. You need to understand how models actually work, including architectures, training dynamics, embeddings, retrieval, evaluation, and the behavior and failure modes of agents that plan and call tools. And you need to be equally fluent in the systems side: distributed computing, large scale training and batch systems, API and SDK design, observability, and infrastructure that holds up at scale.
You should also be AI native in how you build: comfortable pairing with coding agents, LLM powered tooling, and automated evaluation to design and ship the platform itself faster.
Essential functions
- Be involved in designing, build, and operate core components of the AI Engineering Platform: training and experimentation infrastructure, the Batch Platform, agent development frameworks, and the unified AI Studio developer experience.
- Design, build, and operate the Batch Platform: large scale scheduled and on demand batch scoring, offline modelexecution, and high throughput data pipelines for training and feature generation.
- Build self service APIs, SDKs, and tooling that let engineers and data scientists across Visa train, fine tune, deploy,evaluate, and monitor models and agents without platform team intervention.
- Design and extend agent runtimes and orchestration primitives, including tool calling, memory, planning, and multiagent coordination, for agents that operate safely and predictably in production.
- Apply core ML and deep learning knowledge (model architectures, embeddings, fine tuning, evaluationmethodology) to platform design decisions, not just infrastructure decisions
Maintain infrastructure that supports large scale distributed training, high throughput batch processing,and efficient GPU and compute cluster utilization. - Be capable of cost and efficiency optimizations for training and batch workloads, such as distributed scheduling, caching, quantization, and compute right sizing.
- Embed governance, responsible AI, audit, and monitoring capabilities directly into platform components, includingdrift, hallucination, and anomaly detection for agentic systems.
- Partner with the AI Context Platform, AI Runtime Services, and AI Governance Platform teams to deliver acoherent, end to end platform experience.
- Stay current with AI research and the broader platform/tooling ecosystem, and bring in techniques and patternsbefore they become standard practice.
- Use AI coding agents and LLM powered developer tools as part of your own workflow to design, build, and ship AIStudio capabilities faster, treating AI assisted engineering as a core skill, not a side habit.
Qualifications
Basic Qualifications:
- 2+ years of relevant work experience and a Bachelors degree, in Computer Science, Engineering, or a related technical field, OR 5+ years of relevant work experience
- Professional experience in software engineering, ML engineering, or platform/infrastructure engineering.
- Good understanding of core machine learning and deep learning concepts: model training, evaluation, transformer/foundation model architectures, and embeddings.
- Strong systems and software engineering fundamentals: distributed systems, API/SDK design, and cloud infrastructure (e.g., Kubernetes and a major cloud provider such as AWS, Azure, or GCP).
- Proficiency in at least one language commonly used in AI/platform engineering (e.g., Python, Java, Go).A
- Hands on fluency with modern AI assisted development, using coding agents, copilots, and LLM based tooling to accelerate day to day engineering work.
Preferred Qualifications:
- 4 or more years of relevant work experience.
- Experience building with or on top of foundation models / large language models: prompting, fine tuning, retrieval augmented generation (RAG), and evaluation frameworks.
- Experience designing or operating AI agent systems: tool calling, multi agent orchestration, memory systems, or agent frameworks.
- Experience building internal developer platforms, SDKs, or self service tooling used by other engineering or data science teams.
- Experience with feature stores, embedding/vector systems, or knowledge graph and memory systems used by ML or agentic applications.
- Experience with training and batch efficiency techniques such as distributed scheduling, quantization, caching, or compute right sizing at scale.
Experience operating production systems at high scale, such as large scale distributed training jobs or high volume batch processing pipelines. - Experience implementing AI governance, responsible AI, or model risk/compliance controls within a platform.
Visa requires at least 3 days in office, expectations of these days will be confirmed by your Hiring Manager.
Information for US Applicants
For roles located in the US, the estimated salary range for this position is $123,400.00 to $ 191,100.00 USD per year, which may include potential sales incentive payments (if applicable). Salary may vary depending on job-related factors which may include knowledge, skills, experience, and location. In addition, this position may be eligible for bonus and equity.Visa has a comprehensive benefits package for which this position may be eligible that includes Medical, Dental, Vision, 401(k), FSA/HSA, Life Insurance, Paid Time Off, and Wellness Program.
Work Hours
Varies upon the needs of the department.
Travel Requirements
This position requires travel 5-10% of the time.
Mental/Physical Requirements
This position will be performed in an office setting. The position will require the incumbent to sit and stand at a desk, communicate in person and by telephone, frequently operate standard office equipment, such as telephones and computers.
Visa is an EEO Employer
Qualified applicants will receive consideration for employment without regard to race, color religion, sex, national origin, sexual orientation, gender identity, disability or protect veteran status. Visa will also consider for employment qualified applicants with criminal histories in a manner consistent with the EEOC guidelines and applicable local law.