Senior Staff Machine Learning Engineer Platform Identity Jobs

Why Sony Interactive Entertainment? Sony Interactive Entertainment isn’t just the Best Place to Play — it’s also the Best Place to Work. Sony Interactive Entertainment (SIE) is the company behind the PlayStation brand. As a subsidiary of Sony Group Corporation, we’re part of a proud legacy of innovation and excellence. SIE is a dynamic technology company, delivering cutting-edge hardware and network services to more than 100 million people and an entertainment leader, home to some of the most beloved and recognizable intellectual properties (IP) in the world. Our role at SIE is to create and nurture the experiences under the PlayStation brand, a name synonymous with entertainment excellence and creativity. About the Advanced Technology Group The Advanced Technology Group (ATG) is a part of Foundational Systems and Experiences Engineering (FSEE) within the Platform Business Group. Established over 20 years ago as part of PlayStation®’s first-party game development studios, ATG has always focused on helping game developers get the most out of PlayStation® hardware - a mission that continues to this day. As a team, we are responsible for large parts of the graphics software stack used by all PlayStation® games, and collaborate with leading engineering teams worldwide, from game developers to hardware vendors, to deliver key platform technologies. What you’ll be doing You will be working on the current and future rendering APIs used by PlayStation games, as well as the hardware underpinning them. You will design and implement features that enable efficient game development, push GPU programming forward, and create a healthy platform. You will observe industry trends, understanding of – and empathy for – our customers, and business needs as the foundation of your decision making. You will help shape the pat

Biohub is the first large-scale initiative bringing frontier AI models, massive compute, and frontier experimental capabilities under one roof. We're building a general-purpose system to accelerate scientific discovery, integrating frontier AI models, biological foundation models, and lab capabilities, with the ultimate goal of curing disease. Our technology powers scientists around the world, translating AI capabilities into tools that accelerate research everywhere. The Team Biohub is a 501(c)(3) biomedical research organization building the first large-scale scientific initiative combining frontier AI with frontier biology to solve disease. We build the technology to help scientists around the world use AI-powered biology to study how cells operate, organize, and work as part of systems to understand why disease happens and how to correct it. With our compute capacity, AI research and engineering, and state-of-the-art technology for measuring, imaging, and programming biology, we are enabling scientists worldwide to use AI-powered biology to advance our understanding of human health. The Opportunity The role is part of the Data Engineering team, which focuses on owning the strategy, sourcing and implementation for data supporting AI research and development. Our goal is to maximize the speed, agility, and capability of biological AI research by connecting public data resources and Biohub's experimental platforms to AI systems. The data that trains biological frontier models comes in dozens of modalities (sequences, images, spatial coordinates, time series, molecular structures, metadata, publication artifacts) each with its own noise characteristics, biases, and information content. The question of how to represent this data for learning is one of the most important open problems in biological AI. As a Senior Staff Data Engineer at Biohub, you'll be designing systems that ingest data from public repositories, transform heterogeneous biological formats into AI-rea

About the Team Our DD Labs team builds real-time autonomous delivery systems. The Planning & Decision-Making group is investing heavily in deep reinforcement learning to move beyond classical planning, learning policies that generalize across novel driving scenarios, handle long-tail edge cases, and improve continuously from large-scale fleet data. Our models jointly handle prediction and planning in a single unified architecture. Our stack is pure JAX end-to-end: the same code you train with is the code that runs on the robot. No C++ rewrites, no TensorRT export. A new policy goes from training to on-vehicle deployment in minutes. About the Role As a Senior/Staff Deep RL Engineer, you will design, train, and deploy deep reinforcement learning policies that make real-time driving decisions for our autonomous vehicles. You will own the full lifecycle, from problem formulation and reward design through large-scale distributed training to on-vehicle inference. You'll help define how learned components compose with the rest of the autonomy stack to produce robust, shippable behavior. You’re excited about this opportunity because you will… Formulate complex driving tasks as RL problems with well-shaped reward functions and expressive state/action representations. Design and train model-based deep RL agents using GPU-accelerated simulation at massive scale, including improving the simulator itself. Build and maintain distributed training infrastructure in JAX across large compute clusters. Build agentic optimization systems that automatically improve code, run experiments, analyze metrics, and iterate on RL policies with minimal human intervention. We’re excited about you because… BS/MS/PhD in CS, EE, Robotics, or a related field, with a strong foundation in reinforcement learning and deep learning. You have proficiency in using AI coding tools (e.g., Claude Code, Codex, Cursor) in the full software development lifecycle, including designing, generating code, testing,

About Pinecone Pinecone is the knowledge infrastructure for AI at scale. Its leading vector database and knowledge engine, Pinecone Nexus, power accurate, performant AI applications for more than 9,000 customers and 800,000 developers worldwide. Pinecone's mission is to make AI knowledgeable. Pinecone is based in New York and raised $138M in funding from Andreessen Horowitz, ICONIQ, Menlo Ventures, and Wing Venture Capital. About the Team and Role: The Experience team is at the center of one of the most exciting transitions in software development history — the shift from human-driven to agent-driven product experiences. We own Pinecone's API, clients, authentication, revenue, and observability systems, and right now that means redesigning all of it for a world where AI agents are first-class users alongside humans. This is a wide-scope role. You'll own things end-to-end — from backend architecture to API design to SDK and web surfaces. You’ll be working closely with product, design, and other engineering teams to identify user needs and build the right thing, at the right abstraction level, at the right time. Along the way, you will be building high-leverage platform capabilities that accelerate Pinecone’s product development and user growth systems. We're looking for an engineer who sees this moment for what it is: a rare opportunity to shape how developers and agents interact with a category-defining product. You're not waiting to see how the industry figures out MCP, agentic workflows, and AI-native interfaces — you're already experimenting, already forming opinions, already building. You know that speed and leverage matter more than labor, and you've internalized AI-assisted development not as a productivity trick but as a fundamentally different way of working. Responsibilities: Pioneer our agent experience. Shape how AI agents interact with Pinecone — designing interfaces, protocols (MCP), and tooling that make Pinecone the easiest and most capable platform f

Who we are At OKX, we believe that the future will be reshaped by crypto, and ultimately contribute to every individual's freedom. OKX is a leading crypto exchange, and the developer of OKX Wallet, giving millions access to crypto trading and decentralized crypto applications (dApps). OKX is also a trusted brand by hundreds of large institutions seeking access to crypto markets. We are safe and reliable, backed by our Proof of Reserves. Across our multiple offices globally, we are united by our core principles: We Before Me, Do the Right Thing, and Get Things Done. These shared values drive our culture, shape our processes, and foster a friendly, rewarding, and diverse environment for every OK-er. OKX is part of OKG, a group that brings the value of Blockchain to users around the world, through our leading products OKX, OKX Wallet, OKLink and more. About the Team The AI Engineering team is responsible for integrating different AI models with different business lines. You'll need to understand how these models function in diverse domains and develop efficient integration solutions. This team is building the company's crawler platform, supporting multiple areas such as Compliance, Trading, Financial Products, and Business Intelligence. What You’ll Be Doing Understand different AI models’ feature Understand different business domains’ feature Provide integration solutions for AI models into the business domain Maintain systems’ high availability and high throughput Improve technology best practices and code reviews for peers Build and maintain various AI infrastructure / platform What We Look For In You Bachelors in Science / Technology / Engineering / Mathematics Experience in AI Agent / RAG engineering solutions Experience in Microservices / API development with Java/SpringBoot/SpringCloud/Python/Langchain Knowledgeable in multi-threading and distributed architecture. Understanding of mainstream messaging frameworks including Kafka etc. Knowledge of different d

Our vision is to transform how the world uses information to enrich life for all. Join an inclusive team passionate about one thing: using their expertise in the relentless pursuit of innovation for customers and partners. The solutions we build help make everything from virtual reality experiences to breakthroughs in neural networks possible. We do it all while committing to integrity, sustainability, and giving back to our communities. Because doing so can fuel the very innovation we are pursuing. As a key leader within High Bandwidth Memory (HBM) Product Engineering Media Health team, you will spearhead and develop a high performing team driving Media Health Manufacturing and activities for Micron’s latest HBM products, both on DRAM and Interface die and a stacked product. You will lead a portfolio of Best-In-Class Next Generation HBM Product focused on the critical KPI’s Quality, Cost, Cycle Time and Scale. This include lead the manufacturing test flow development and validation, improve the device/stacked yield with most optimized extrinsic and intrinsic reliability through in depth circuit debug, test coverage optimization and strategy innovation, test time improvement for manufacturing flow and drive the DFT (design for test) strategy to stage new ideas to improve cost, cycle time and quality. You will also collaborate with cross functional team (Fab, Product Leads, Design, Technology Development, Design, Test Solution Engineering etc) with diverse expertise to achieve both strategic and tactical objectives and stay ahead on technology and product leadership. The future of HBM are exciting and dynamics. Micron is seeking experienced individuals that find technical challenges engaging and invigorating. Responsibilities: Optimized Test

Who we are About Stripe Stripe is a financial infrastructure platform for businesses. Millions of companies—from the world's largest enterprises to the most ambitious startups—use Stripe to accept payments, grow their revenue, and accelerate new business opportunities. Our mission is to increase the GDP of the internet, and we have a staggering amount of work ahead. That means you have an unprecedented opportunity to put the global economy within everyone's reach while doing the most important work of your career. About the team The Merchant Experience team builds the UI operating system and core experiences that power decades of growth for millions of businesses globally, enabling Stripe to be the business SaaS solution of choice for next-gen companies. We own the foundational platforms and product surfaces that merchants interact with every day - from the Stripe Dashboard to the frameworks and tooling that hundreds of Stripe engineers use to build world-class user interfaces. We're at an inflection point. AI is fundamentally changing how users interact with software, and we believe Stripe is uniquely positioned to reimagine what a merchant experience looks like in an AI-native world. At the same time, we're raising the bar on the quality and performance of everything we ship - because the millions of businesses that depend on Stripe deserve an experience that is fast, reliable, and crafted with care. What you'll do As a Senior Staff Engineer on the Merchant Experience team, you'll be a technical leader across 50+ Engineers responsible for shaping the next generation of how Stripe merchants interact with our products. You'll drive a step-function improvement in the quality and performance of our core surfaces, and help define what AI-powered merchant experiences look like at Stripe. Responsibilities Define technical strategy for merchant-facing experiences across Stripe, with a focus on quality, performance, and the integration of AI into core user workflows Lead t

Who We Are At OKX, we believe that the future will be reshaped by crypto, and ultimately contribute to every individual's freedom. OKX is a leading crypto exchange, and the developer of OKX Wallet, giving millions access to crypto trading and decentralized crypto applications (dApps). OKX is also a trusted brand by hundreds of large institutions seeking access to crypto markets. We are safe and reliable, backed by our Proof of Reserves. Across our multiple offices globally, we are united by our core principles: We Before Me, Do the Right Thing, and Get Things Done. These shared values drive our culture, shape our processes, and foster a friendly, rewarding, and diverse environment for every OK-er. OKX is part of OKG, a group that brings the value of Blockchain to users around the world, through our leading products OKX, OKX Wallet, OKLink and more. About the Team We sit on the front lines of OKX's security and compliance mission. Our charter is to build the flexible, scalable, and intelligent AI systems that detect and prevent identity fraud, account takeover, and large-scale organized abuse — in real time and at the scale of a global financial platform. We own the core identity-risk capabilities that decide, millions of times a day, whether an account can be trusted: identity trust signals, behavioral risk modeling, relationship and cluster intelligence, and face, liveness and document verification. We work shoulder-to-shoulder with Product, Risk, Compliance, and Engineering to turn the hardest adversarial problems in crypto into production-grade defenses that protect good users and the open financial system. About the Role We are looking for a Senior AI Engineer to design, build, and own key identity-risk and trust models at OKX. You will build models that combine identity verification results, device and geo data, and user behavior into trust and risk signals, and use graph and clustering methods to uncover organized, industrialized fraud. This is an

Who We Are At OKX, we believe that the future will be reshaped by crypto, and ultimately contribute to every individual's freedom. OKX is a leading crypto exchange, and the developer of OKX Wallet, giving millions access to crypto trading and decentralized crypto applications (dApps). OKX is also a trusted brand by hundreds of large institutions seeking access to crypto markets. We are safe and reliable, backed by our Proof of Reserves. Across our multiple offices globally, we are united by our core principles: We Before Me , Do the Right Thing , and Get Things Done . These shared values drive our culture, shape our processes, and foster a friendly, rewarding, and diverse environment for every OK-er. OKX is part of OKG, a group that brings the value of Blockchain to users around the world, through our leading products OKX, OKX Wallet, OKLink and more. Key Responsibilities Design and optimize high throughput order execution systems with sub-millisecond latency targets Build real time market data pipelines aggregating order books and price feeds from multiple exchanges Implement position tracking with exchange reconciliation, risk limits, and multi account support Develop exchange connectivity using FIX 5.0+, WebSocket, and HTTP/2 protocols Profile and eliminate latency hotspots using garbage free coding and lock free concurrency Implement pre-trade risk controls, circuit breakers, and rate limiters Qualifications Solid STEM (Science/Technology/Engineering/Mathematics) education background. 5-8 years software development experience, preferably in trading/oms/latency sensitive systems Strong Java 17+ proficiency with understanding of JVM internals and GC behavior Hands-on experience with FIX 5.0+, WebSocket, and HTTP/2 protocols Solid multi-threaded programming and concurrent systems knowledge Understanding of order lifecycle and market microstructure basics Strong problem solving skills and attention to detail Excellent co

Join us in building the future of finance. Our mission is to democratize finance for all. An estimated $124 trillion of assets will be inherited by younger generations in the next two decades. The largest transfer of wealth in human history. If you’re ready to be at the epicenter of this historic cultural and financial shift, keep reading. About the team + role We are building an elite team, applying frontier technologies to the world’s biggest financial problems. We’re looking for bold thinkers. Sharp problem-solvers. Builders who are wired to make an impact. Robinhood isn’t a place for complacency, it’s where ambitious people do the best work of their careers. We’re a high-performing, fast-moving team with ethics at the center of everything we do. Expectations are high, and so are the rewards. The Data Platform organization builds and operates the systems that power how data is stored, moved, and consumed across Robinhood. This organization spans three core pillars: Storage (Postgres, DynamoDB, and caching systems), Streaming (real-time event infrastructure), and Data Lake (ingestion and compute systems built on Delta Lake). Together, these platforms support transactional workloads, real-time data processing, and large-scale analytics that are critical to Robinhood’s products and operations. The team owns the full lifecycle of data—from low-latency order path systems to near real-time and batch analytics—serving millions of users and internal teams across the company. As a Senior Staff Software Engineer , you will serve as the technical lead across the Data Platform organization, shaping architecture and guiding execution across multiple teams. You’ll work on complex distributed systems challenges such as database sharding and proxy architectures, real-time streaming and CDC systems, and large-scale data ingestion and compute platforms. You’ll define and drive key technical bets, partner with engineering leaders to align platform capabilities with business needs,

Our vision is to transform how the world uses information to enrich life for all. Join an inclusive team passionate about one thing: using their expertise in the relentless pursuit of innovation for customers and partners. The solutions we build help make everything from virtual reality experiences to breakthroughs in neural networks possible. We do it all while committing to integrity, sustainability, and giving back to our communities. Because doing so can fuel the very innovation we are pursuing. Responsibilities: Reliability Test Flow Development: Define and develop reliability stress, test flows and test plans to cover all Product Reliability aspects. Reliability Test Plan and Execution: Work closely with Global Quality team to define qualification plan, review and manage the qualification execution progress, and drive for qualification gating issue resolution. DPM reduction : Drive down and achieving Extrinsic Reliability (Time 0 and Field DPM) and Intrinsic Reliability through Optimized Manufacturing Test Flows and Test Strategy to meet critical KPI’s Quality, Cost and Cycle Time Root Cause and Resolution of Qual and RMA Device Issues pertaining to defectivity and intrinsic reliability: Debug and identify root cause and failures in reliability tests by electric failure analysis(EFA) and Physical Failure Analysis (PFA) and drive for resolution and improvements through cross functional team collaboration. Process Conversions and HVM: The role involves providing recommendation to fab teams for new process conversions to reduce cost, increase yields. This is critical as it directly impacts the yield, quality, reliability, and performance of HBM products, which are key components in many modern technologies.</li

Senior/Staff HBM PYE Product Development Failure Analysis Engineer — Boise, ID - Main Site. Apply via Workday.

Senior/Staff/Principal Engineer, HIG-HBM Product System & Engineering (Media Health Reliability Coding) — Fab 10A, Singapore. Apply via Workday.

Senior/Staff/Principal Engineer, HIG-HBM Product System & Engineering (Media Health Manufacturing) — Fab 10A, Singapore. Apply via Workday.

About Druva Druva is the resilience foundation for the AI enterprise, helping organizations secure and recover from connected risk across data, cyber, identity, and AI. The Resilience Cloud is a fully managed, cloud-native SaaS platform that delivers air-gapped and immutable protection across cloud, SaaS, on-premises, endpoint, and edge environments. Powered by Dru MetaGraph, Druva’s graph-powered intelligence layer, the platform connects critical business context so customers can understand risk, respond faster, recover cleanly, and govern data with greater confidence. Trusted by nearly 7,500 customers, including 75 of the Fortune 500, Druva helps safeguard the critical information and systems businesses depend on in an increasingly connected world. Visit druva.com and follow us on LinkedIn , X and Facebook . The Data team within Foundation processes data at scale. It builds and operates tools that scan petabytes of data to extract insights. The team builds indices and views for semi-structured data to facilitate analytics, using both in-house and off-the-shelf tools. . The team diligently keeps track of newer services, storage tiers, and various aspects of existing AWS services to take advantage of the continuous evolution of services and use them effectively in the background. As a Sr. Software Engineer, you will be providing technical leadership to create high-quality software by owning low level , design and implementation of services within a product. This role will require excellent communication skills as you will collaborate with Product Management to refine requirements, product architects to propose design changes, and other product owners to drive features to completion with good quality. Key Skill AI first mindset to software development, having experience using genAI during various phases of software development lifecycle from design to code to test using tools like 'cursor' 5-7