Tech jobs
Job listings
Staff AI Engineer
Lead AI platform development, architecting and deploying production-grade LLM/RAG systems, MLOps pipelines, and distributed AI services for a fintech company.
Principal AI/ML Research Engineer
Principal AI/ML Research Engineer leads applied research in generative AI, deep learning, and Transformers to build Payment Foundation Models for commerce and fintech challenges, driving novel architectures from prototype to production.
Senior Software Engineer, CUDA Python Core Libraries
Build Python APIs and bindings for CUDA Core Libraries, integrating C/C++/Rust to accelerate GPU-accelerated software for HPC and AI workloads.
Amazon Software Development Manager, Compiler
- 5+ years of engineering team management experience - 9+ years of working directly within engineering teams experience - 4+ years of designing or architecting (design patterns, reliability and scaling) of new and…
Amazon Software Development Manager, Machine Learning Accelerators
Lead a team of compiler engineers to design and optimize ML compiler solutions for AWS Neuron, driving hardware bring-up and influencing chip design for Trainium-based ML accelerators.
Amazon Senior Machine Learning Compiler Engineer
Build and optimize the Neuron compiler to transform ML models (PyTorch, TensorFlow, JAX) for AWS Inferentia/Trainium chips, solving compiler passes and performance tuning for large language and diffusion models.
Amazon Senior Machine Learning Compiler Engineer
Build and optimize a deep-learning compiler stack (AWS Neuron) that converts PyTorch/TensorFlow models into high-performance code for AWS’s custom ML accelerators (Inferentia/Trainium).
Amazon Machine Learning Performance Engineer
Optimize AWS Neuron’s ML software stack for performance on custom Inferentia/Trainium accelerators, writing high-performance kernels and collaborating across compiler, frameworks, and hardware teams.
AI Engineer
Design and build AI/ML models to solve real-world problems, using Python and frameworks like PyTorch or TensorFlow, while collaborating in a hybrid office setup.
Amazon Senior Machine Learning Software Engineer
Build and optimize low-level compute kernels and inference pipelines for large language models running on custom ML hardware, integrating with frameworks like PyTorch and vLLM.
Amazon Machine Learning Software Engineer, Data Plane
Builds and optimizes low-level compute kernels and serving infrastructure for large language model inference on custom ML hardware, integrating frameworks like PyTorch and vLLM.
Amazon Senior Software Development Engineer, Edge AI ML Platform
Build and lead the Edge AI ML platform that trains, optimizes, and deploys large generative models on devices and in the cloud using PyTorch, TensorFlow, and Kubernetes.
AI Engineer (Remote)
Build and deploy AI models, RAG pipelines, and LLM-powered systems for early-stage startups backed by a VC firm.
Amazon Senior Applied Scientist, Machine Learning
Senior Applied Scientist designing and deploying ML compiler analyzers and tooling to optimize deep learning models across frameworks like PyTorch and TensorFlow, working with custom ML accelerators like Inferentia and Trainium.
Amazon Machine Learning Software Engineer, Data Plane
Build and optimize low-level compute kernels and serving integrations for large language model inference on custom ML hardware, spanning model execution, memory management, and distributed systems.
Amazon Software Development Engineer, AWS SageMaker AI
Build and operate the distributed orchestration engine for Amazon SageMaker AI’s Model Factory, running large-scale LLM training and customization workflows across thousands of GPUs and Trainium devices.
Amazon Senior AI/ML Specialist Solutions Architect
Design and advise customers on generative AI and ML solutions using AWS services like Amazon Bedrock and SageMaker, including RAG pipelines and agentic workflows.
Data Scientist
Senior Data Scientist designs and deploys AI/ML models, builds real-time data pipelines, and creates dashboards using Elastic, Kafka, Hadoop, Tableau, and Power BI to extract business insights.
Machine Learning Software Engineer (Remote)
Build and ship ML infrastructure for autonomous freight systems: multimodal data pipelines, model training/evaluation workflows, and low-latency inference tooling using PyTorch/TensorFlow/JAX.
Manager,Database
About Affinity Affinity is pioneering new frontiers in AdTech: developing solutions that push past today’s limits and open new opportunities. We are a global AdTech company helping publishers discover better ways to…