Flash Attention Jobs - Remote & On-site

Flash Attention jobs hiring now at vetted startups and product companies on Alion, updated daily. Average salary around $342,083/year. Compare remote, hybrid and on-site positions with relocation and visa sponsorship. Browse and apply today.

You searched for “Flash Attention Jobs - Remote & On-site”, so this list shows: technical roles only · the stack names Flash Attention · known staffing agencies left out. Adjust the filters

AI/ML
Data Science
Backend
Frontend
Mobile
DevOps
Web3
Games
Hardware
Robotics
Security
QA
Leadership
Networking
Product
Design
Analytics
Support
Solutions
Enterprise Apps
Quantum

NVIDIA

nvidia.com
nvidia.com • HQ: Santa Clara, United States • Computer Components • Servers & Data Center Hardware • AI Chips & Accelerators • Processors, MCUs & AI Chips • 5000+ employees • Est. 1993 • Trust A • Hiring 2115, 186 new
HQ: Santa Clara, United States • Computer Components • Servers & Data Center Hardware • AI Chips & Accelerators • Processors, MCUs & AI Chips • 5000+ employees • Est. 1993 • Trust A • Hiring 2115, 186 new
NVIDIA is an American technology company founded in 1993 that invented the graphics processing unit and has become the dominant supplier of accelerated computing platforms for artificial intelligence. Its portfolio spans data centre GPUs and systems built on the Hopper and Blackwell architectures, GeForce consumer graphics, automotive and robotics platforms, high-speed networking acquired with Mellanox, and the CUDA software stack that binds the ecosystem together. Headquartered in Santa Clara, California, the company sells to cloud providers, enterprises, research institutions and gamers worldwide and is one of the most valuable listed businesses on the Nasdaq.
Jensen Huang · Co-Founder, President & Chief Executive Officer · decision-maker
Jay Puri · Executive Vice President, Worldwide Field Operations · decision-maker
Colette Kress · Executive Vice President & Chief Financial Officer · decision-maker
Curtis Priem · Co-Founder & former Chief Technical Officer · founder
Careers mailbox is known
Verified live · 1 hour ago 6 days ago Likely open · 60%

Software Engineer, DL Kernel Libraries - New College Grad 2027

$108k – $178k per year • In office (Santa Clara, United States) • Bachelor's Degree • Full-Time
Python
C++
C++
TensorFlow C++
PyTorch C++
AI/ML
Flash Attention
vLLM
CUDA Toolkit
JAX
SGLang
ONNX
TensorFlow
PyTorch
LLM
CUDA
Triton
cuDNN
MLIR
Apache TVM
XLA
CUTLASS
Machine Learning
+13
Apply

Speechmatics

speechmatics.com
speechmatics.com • HQ: Cambridge, United Kingdom • Machine Learning • Speech & Audio AI • 201-500 employees • Est. 2006 • Hiring 14
HQ: Cambridge, United Kingdom • Machine Learning • Speech & Audio AI • 201-500 employees • Est. 2006 • Hiring 14
Speechmatics is a British company founded in 2006 that builds speech recognition designed to work across accents and dialects. It trains largely with self-supervised learning on unlabelled audio, which improved accuracy for speakers that mainstream systems transcribe poorly. Its engine is licensed by broadcasters, contact centres and software vendors and runs in the cloud or on-premise.
No named decision-maker yet.
Careers mailbox is known
Verified live · 3 hours ago 14 days ago Hiring now · 90%

Senior Machine Learning Engineer

≈ $100k – $186k per year (Estimated) • Senior • Cambridge • London • Remote (United Kingdom) • Senior
Speech Recognition
Self-Supervised Learning
Machine Learning
Flash Attention
Speculative Decoding
Cybersecurity
GDPR
Apply
Verified live · 3 hours ago 14 days ago Hiring now · 90%

Machine Learning Team Lead

≈ $105k – $194k per year (Estimated) • Lead • London • Cambridge • Remote (United Kingdom) • Staff
Python
AI/ML
Speech Recognition
Self-Supervised Learning
Machine Learning
Flash Attention
TensorFlow
PyTorch
Speculative Decoding
Cybersecurity
GDPR
+1
Apply

Akrys.ai

akrys.ai
akrys.ai • HQ: Bengaluru, India • Document AI • Speech & Audio AI • EHR & Clinical Software • Medical AI • Est. 2025 • Likely startup • Hiring 5
HQ: Bengaluru, India • Document AI • Speech & Audio AI • EHR & Clinical Software • Medical AI • Est. 2025 • Likely startup • Hiring 5
Akrys AI is a healthtech company building a clinical intelligence platform to convert doctor-patient conversations into structured records and clinical notes, offering AI-assisted support for healthcare workflows.
No named decision-maker yet.
Careers mailbox is known
16 days ago Likely open · 60%

Lead AI Engineer - Speech & Clinical AI

≈ $25k – $63k per year (Estimated) • Lead • 6+ years exp • In office (Bengaluru, India) • Staff
Python
SQL
Bash
Rust
C++
Python
Dask
C++
PyTorch C++
TensorFlow C++
AI/ML
Weights & Biases
Qwen
Spark
Bitsandbytes
DeepSpeed
LoRA
vLLM
MLFlow
Triton Inference Server
Gemma
Fine-tuning
RLHF
Quantization
Scikit-learn
Prompt Engineering
SciPy
Knowledge Distillation
AI Agents
NLP
NER
Speech Recognition
AWQ
GPTQ
ONNX
TensorRT
PEFT
QLoRA
TGI
TRL
Llama
Mistral
Accelerate
Kaldi
Transformers
NumPy
PyTorch
LLM
RAG
Whisper
Ray
Hallucination
Synthetic Data
Encoder-Decoder
Hugging Face
DPO
SFT
PPO
Pre-training
FSDP
NVIDIA NeMo
TorchServe
NCCL
Model Distillation
Flash Attention
JAX
xFormers
TensorFlow
DevOps
GCP
SLURM
Azure
CI/CD
Git
AWS
Docker
Kubernetes
SOAP
Cybersecurity
GDPR
HIPAA
+68
Apply

Tether

tether.to
tether.to • HQ: San Salvador, El Salvador • Crypto Payments • Cryptocurrencies • Stablecoins • 1001-5000 employees • Est. 2014 • Trust A • Hiring 185, 54 new
HQ: San Salvador, El Salvador • Crypto Payments • Cryptocurrencies • Stablecoins • 1001-5000 employees • Est. 2014 • Trust A • Hiring 185, 54 new
Tether is a digital asset company founded in 2014 that issues USDT, the largest stablecoin in circulation and the most heavily traded asset in crypto markets. Each token is intended to hold a one-to-one peg with the United States dollar and is backed by a reserve portfolio dominated by short-dated Treasury bills, with regular attestation reports published on the reserve composition. Headquartered in San Salvador, El Salvador, the group has expanded beyond stablecoin issuance into gold-backed tokens, Bitcoin mining, energy, peer-to-peer communications and local artificial intelligence infrastructure.
Giancarlo Devasini · Chairman · decision-maker
Paolo Ardoino · Chief Executive Officer · decision-maker
Craig Sellars · Co-Founder · founder
Brock Pierce · Co-Founder · founder
No careers mailbox known
Verified live · 2 hours ago 18 days ago Hiring now · 90%

AI Research Engineer (Kernel & Inference Optimization) - 100% Remote Worldwide

Andorra la Vella • Remote (Andorra) • English: B2
Flash Attention
Quantization
Multimodal AI
Diffusion Models
NLP
Transformers
Tokenization
Speculative Decoding
KV Cache
Machine Learning
Mobile
Metal
Web3
Bitcoin
Management
Telegram
WhatsApp
+6
Apply
Verified live · 2 hours ago 5 months ago Long shot · 15%

AI Research Engineer (Model Compression & Quantization) - 100% Remote Worldwide

Remote (United Kingdom) • English: B2
Flash Attention
Quantization
Multimodal AI
Diffusion Models
NLP
Transformers
Tokenization
Speculative Decoding
KV Cache
Machine Learning
Mobile
Metal
Web3
Bitcoin
Management
Telegram
WhatsApp
+6
Apply

Alibaba Group

alibaba.com
alibaba.com • HQ: Hangzhou, China • Logistics • E-commerce Platforms • Foundation Models • Cloud Platforms (IaaS & PaaS) • Marketplaces • 1001-5000 employees • Est. 1999 • Trust A • Hiring 2180, 274 new
HQ: Hangzhou, China • Logistics • E-commerce Platforms • Foundation Models • Cloud Platforms (IaaS & PaaS) • Marketplaces • 1001-5000 employees • Est. 1999 • Trust A • Hiring 2180, 274 new
Alibaba Group is a Chinese technology conglomerate founded in Hangzhou in 1999 by Jack Ma and seventeen colleagues, built around commerce and cloud computing. Its domestic marketplaces Taobao and Tmall handle a large share of Chinese retail, while Alibaba.com, AliExpress, Lazada and Trendyol serve cross-border and international buyers, supported by the Cainiao logistics network. Alibaba Cloud is the largest public cloud provider in Asia and develops the open-weight Qwen model family, which has become one of the most downloaded foundation model series in the world.
Eddie Wu · Chief Executive Officer · decision-maker
Joseph Tsai · Co-Founder & Chairman · decision-maker
Toby Xu · Chief Financial Officer · decision-maker
Jack Ma · Founder · founder
No careers mailbox known
Verified live · 3 hours ago 19 days ago Hiring now · 74%

阿里云智能-技术服务工程师-AI全栈-北京

Junior • 2+ years exp • In office (Beijing, China) • Junior • Chinese: C1
Python
Databases
Apache Kafka
AI/ML
LangChain
Dify
Flash Attention
LlamaIndex
LoRA
vLLM
CUDA Toolkit
Triton Inference Server
RLHF
Prompt Engineering
AI Agents
SGLang
AWQ
GPTQ
LLaMA-Factory
TensorRT-LLM
TGI
LLM
RAG
CUDA
SFT
NCCL
NVLink
KV Cache
TensorRT
PEFT
DevOps
Kubernetes
AIOps
+22
Apply
a-star.edu.sg • HQ: Singapore • Research Institutes • 1001-5000 employees • Est. 1991 • Trust B • Hiring 438, 17 new
HQ: Singapore • Research Institutes • 1001-5000 employees • Est. 1991 • Trust B • Hiring 438, 17 new
The Agency for Science, Technology and Research is a leading public sector organization dedicated to advancing economic-oriented scientific discovery and technological innovation. Headquartered in Singapore, the agency oversees numerous specialized research institutes spanning biomedical sciences, physical sciences, and engineering. By bridging the gap between academia and industry, it fosters public-private partnerships, cultivates scientific talent, and propels the growth of the regional innovation ecosystem.
No named decision-maker yet.
No careers mailbox known
≈ $41k – $104k per year (Estimated) • In office (Singapore)
Verilog
C++
SystemVerilog
AI/ML
Flash Attention
CUDA Toolkit
Quantization
LLM
Mixture of Experts
CUDA
vLLM
TensorRT
Triton
+4
Apply

IDDA

idda.az
idda.az • HQ: Baku, Azerbaijan • Hiring 21, 6 new
HQ: Baku, Azerbaijan • Hiring 21, 6 new
Fəaliyyətimiz. mygov tətbiqinin istifadəçi sayı.
No named decision-maker yet.
No careers mailbox known
Verified live · 1 hour ago 28 days ago Fading · 35%

LLM Fine-Tuning & Evaluation Engineer

In office (Baku, Azerbaijan) • Full-Time
Python
AI/ML
Weights & Biases
DeepSpeed
Flash Attention
LoRA
vLLM
MLFlow
Fine-tuning
Quantization
Prompt Engineering
Multimodal AI
Knowledge Distillation
Axolotl
LLaMA-Factory
PEFT
QLoRA
TRL
Unsloth
Accelerate
Transformers
PyTorch
LLM
RAG
Tokenization
Hallucination
Synthetic Data
Hugging Face
DPO
SFT
Pre-training
FSDP
NCCL
Red Teaming
LLM Guardrails
Speculative Decoding
KV Cache
Model Distillation
Machine Learning
DevOps
SLURM
Git
Docker
Linux
Management
Agile
+35
Apply

Moore Threads

mthreads.com
mthreads.com • HQ: Beijing, China • AI Chips & Accelerators • Processors, MCUs & AI Chips • 1001-5000 employees • Est. 2020 • Trust C • Hiring 250, 13 new
HQ: Beijing, China • AI Chips & Accelerators • Processors, MCUs & AI Chips • 1001-5000 employees • Est. 2020 • Trust C • Hiring 250, 13 new
Moore Threads is a Chinese semiconductor company founded in 2020 by former Nvidia executives to build full-function graphics processors. Its MTT series supports graphics, video and artificial intelligence compute, and the company promotes them as a domestic option for training clusters under export controls. It develops the accompanying driver and framework software to make the chips usable with common AI toolchains.
No named decision-maker yet.
Careers mailbox is known
Verified live · 2 hours ago 2 months ago Long shot · 6%

AI应用推理引擎研发工程师(J10989)

In office (Beijing, China)
llama.cpp
Flash Attention
vLLM
CUDA Toolkit
CUDA
Triton
OCR
ONNX Runtime
MLIR
Apache TVM
CUTLASS
Speculative Decoding
+4
Apply

XPeng

xiaopeng.com
xiaopeng.com • HQ: Guangzhou, China • Autonomous Driving • Automotive Manufacturing • 1001-5000 employees • Est. 2014 • Trust B • Hiring 1001, 37 new
HQ: Guangzhou, China • Autonomous Driving • Automotive Manufacturing • 1001-5000 employees • Est. 2014 • Trust B • Hiring 1001, 37 new
XPeng is a Chinese smart electric vehicle maker headquartered in Guangzhou, Guangdong, developing its own driver-assistance software, AI chips, electric powertrains and in-car operating systems for models such as the P7, G6, G9, X9 and MONA M03. Founded in 2014 by He Xiaopeng and partners, it listed in New York in 2020 and Hong Kong in 2021, passed one million cumulative deliveries by the end of 2025, sells in about 60 countries, and is developing the IRON humanoid robot and flying cars through affiliate Aridge. It hires AI algorithm, autonomous driving, chip and vehicle engineers, product managers, and international sales and finance staff.
No named decision-maker yet.
Careers mailbox is known
Verified live · 3 hours ago 3 months ago Fading · 36%

端侧大模型推理与量化资深工程师

In office (Beijing, China) • Full-Time • Senior • Chinese: C1
C++
AI/ML
Flash Attention
Apply

Modal

modal.com
modal.com • HQ: New York, United States • MLOps • Cloud Platforms (IaaS & PaaS) • AI Compute & Inference • 51-200 employees • Est. 2021 • Trust A • Hiring 42, 3 new
HQ: New York, United States • MLOps • Cloud Platforms (IaaS & PaaS) • AI Compute & Inference • 51-200 employees • Est. 2021 • Trust A • Hiring 42, 3 new
Modal is an AI infrastructure company headquartered in New York City and founded in 2021. It provides a serverless cloud platform featuring sub-second cold starts and instant autoscaling that enables developers to run GPU-accelerated workloads, including model inference and fine-tuning, using a Python-native SDK. The company operates a globally distributed compute network designed for AI applications and serves a diverse range of industries such as generative AI and biotechnology.
Seth Zelnick · Executive Officer · decision-maker
Erik Bernhardsson · Executive Officer, Director · decision-maker
Akshat Bubna · Executive Officer · decision-maker
Paul Butler · Executive Officer, Director · decision-maker
No careers mailbox known
Top 25% payVerified live · 1 hour ago Visa sponsor 3 months ago Likely open · 64%

Member of Technical Staff - Research, Inference

$150k – $350k per year • Staff+ • In office (New York, San Francisco, United States) • Full-Time • Staff
Flash Attention
Quantization
Multimodal AI
SGLang
LLM
Speculative Decoding
KV Cache
Analytics
Seaborn
Matplotlib
+1
Apply

Unitree Robotics

unitree.com
unitree.com • HQ: Hangzhou, China • Robotic Components & Sensors • Robotics AI • Humanoid Robots • Est. 2016 • Hiring 71, 2 new
HQ: Hangzhou, China • Robotic Components & Sensors • Robotics AI • Humanoid Robots • Est. 2016 • Hiring 71, 2 new
Unitree Robotics is a Chinese company founded in 2016 that made legged robots affordable, selling quadrupeds at a fraction of the price of earlier research platforms. Its G1 and H1 humanoids and Go series quadrupeds are widely used by universities, developers and industrial customers, and its demonstrations of agile locomotion have circulated globally. The company designs its own motors, reducers and control software.
No named decision-maker yet.
Careers mailbox is known
Verified live · 3 hours ago 4 months ago Long shot · 5%

AI算法工程师(校招)

In office (Beijing, China) • Chinese: C1
C++
AI/ML
Flash Attention
LoRA
PEFT
Apply
Verified live · 3 hours ago 7 months ago Long shot · 3%

AI算法工程师(实习)

In office (Hangzhou, China) • Internship • Intern • Chinese: C1
C++
AI/ML
Flash Attention
LoRA
PEFT
Apply

Yandex

ya.ru
ya.ru • HQ: Moscow, Russia • Delivery • LLM & Generative AI • Consumer Internet Groups • Cloud Platforms (IaaS & PaaS) • Ride Hailing • Advertising • Search Engines • 5000+ employees • Est. 1997 • Hiring 344, 20 new
HQ: Moscow, Russia • Delivery • LLM & Generative AI • Consumer Internet Groups • Cloud Platforms (IaaS & PaaS) • Ride Hailing • Advertising • Search Engines • 5000+ employees • Est. 1997 • Hiring 344, 20 new
Yandex is a Russian internet company whose search engine launched in 1997 and which grew into the dominant technology group in its home market, running search, advertising, maps, ride hailing, food delivery, e-commerce and cloud services. Its Yandex Go application combines taxi, delivery and logistics into one of the most used consumer services in the country, while Alice and the YandexGPT model family provide its assistant and generative AI layer. Following the 2024 split of the former Yandex N.V., the Russian business was sold to a domestic consortium and now operates independently from Nebius Group, which inherited the international assets.
Artem Savinovsky · Chief Executive Officer · decision-maker
Ilya Segalovich · Co-Founder · founder
Arkady Volozh · Co-Founder · founder
Careers mailbox is known
Senior • Hybrid • Senior • Russian: C1
DeepSeek
CUDA Toolkit
AI Agents
YandexGPT
PyTorch
LLM
CUDA
Triton
Flash Attention
FSDP
+2
Apply

Eloelo

eloelo.in
eloelo.in • HQ: Bengaluru, India • Digital Media • Content Creation • 201-500 employees • Startup • Trust A • Hiring 31, 3 new
HQ: Bengaluru, India • Digital Media • Content Creation • 201-500 employees • Startup • Trust A • Hiring 31, 3 new
Eloelo is an Indian vernacular lifestyle and consumer technology platform with influencer-driven content, live streaming and a skill-learning app called Master. It hires product designers, product managers, data and AI engineers and video editors. The company is headquartered in Bengaluru, India.
No named decision-maker yet.
Careers mailbox is known
Verified live · 19 min ago 5 months ago Long shot · 10%

AI - Engineer

$74k – $84k per year • Middle • 3+ years exp • In office (Bengaluru, India) • Full-Time • Middle
Python
SQL
AI/ML
ChatGPT
Stable Diffusion
Bitsandbytes
Flash Attention
LoRA
Gemma
Fine-tuning
RLHF
Multimodal AI
Diffusion Models
AWQ
ControlNet
DreamBooth
GGUF
GPTQ
PEFT
QLoRA
Mistral
Transformers
LLM
OpenAI
DPO
SFT
Text-to-Speech
LLM Evaluation
Weights & Biases
DeepSpeed
vLLM
CUDA Toolkit
Prompt Engineering
NLP
Axolotl
LLaMA-Factory
TGI
PyTorch
GAN
LLaVA
CUDA
Hugging Face
FSDP
GPT-4
DevOps
GCP
AWS
+37
Apply

Voyager Sopris Learning

voyagersopris.com
voyagersopris.com • Secondary Education • EdTech Platforms • 1001-5000 employees • Trust B • Hiring 37, 5 new
Secondary Education • EdTech Platforms • 1001-5000 employees • Trust B • Hiring 37, 5 new
Voyager Sopris Learning provides educational programs and tools that help educators improve student achievement.
No named decision-maker yet.
No careers mailbox known
Open 164 daysVerified live · 1 day ago 5 months ago Long shot · 11%

Machine Learning Software Engineer II

≈ $109k – $219k per year (Estimated) • Middle • 2+ years exp • United States • Remote (United States) • Full-Time • Middle
Python
Java
C++
C++
TensorFlow C++
PyTorch C++
AI/ML
Flash Attention
Quantization
Scikit-learn
ONNX
TensorFlow
PyTorch
Amazon SageMaker
ONNX Runtime
Machine Learning
LoRA
Fine-tuning
AI Agents
NLP
PEFT
QLoRA
LLMOps
DevOps
CI/CD
AWS
Docker
AWS Lambda
Amazon ECS
Terraform
CloudFormation
+20
Apply

Tahoe Therapeutics

tahoebio.ai
tahoebio.ai • HQ: San Francisco, United States • Molecular Diagnostics • Drug Discovery • 51-200 employees • Est. 2023 • Startup • Trust B • Hiring 3
HQ: San Francisco, United States • Molecular Diagnostics • Drug Discovery • 51-200 employees • Est. 2023 • Startup • Trust B • Hiring 3
Tahoe Therapeutics generates very large single-cell perturbation datasets to train foundation models of how cells respond to drugs. Founded in 2023 in San Francisco, it released the Tahoe-100M dataset openly to the research community. Its models aim to predict drug effects across cancer cell contexts before laboratory testing.
No named decision-maker yet.
Careers mailbox is known
Open 332 daysTop 25% pay Visa sponsor 11 months ago Long shot · 8%

Senior Machine Learning Scientist

$200k – $275k per year • Senior • In office (South San Francisco, United States) • PhD • Senior
JAX
SciPy
Multimodal AI
TensorFlow
Pandas
NumPy
PyTorch
Machine Learning
Flash Attention
Self-Supervised Learning
Contrastive Learning
FSDP
+4
Apply

Nebius

nebius.com
nebius.com • HQ: Amsterdam, Netherlands • Data Centers & Colocation • Cloud Platforms (IaaS & PaaS) • AI Compute & Inference • 1001-5000 employees • Trust B • Hiring 336, 44 new
HQ: Amsterdam, Netherlands • Data Centers & Colocation • Cloud Platforms (IaaS & PaaS) • AI Compute & Inference • 1001-5000 employees • Trust B • Hiring 336, 44 new
Nebius is an international technology company that specializes in building full-stack artificial intelligence infrastructure and high-performance cloud GPU platforms for AI model development. Headquartered in Amsterdam, Netherlands, the firm operates energy-efficient data centers across Europe and North America to provide scalable compute, storage, and software tools for machine learning workloads.
No named decision-maker yet.
Careers mailbox is known
Open 344 daysVerified live · 23 min ago 11 months ago Long shot · 3%

Senior ML Engineer (AI Research)

≈ $127k – $326k per year (Estimated) • Senior • Amsterdam • Remote (Europe, Israel) • Bachelor's Degree • Senior • English (Optional)
Python
AI/ML
Reinforcement Learning
AI Agents
Post-training
Model Distillation
Machine Learning
Flash Attention
Quantization
LLM
DPO
PPO
FSDP
Reward Modeling
DevOps
CI/CD
+6
Apply

Scale AI

scale.com
scale.com • HQ: San Francisco, United States • LLM & Generative AI • AI Evaluation & Observability • AI Training Data & Annotation • 1001-5000 employees • Est. 2016 • Trust B • Hiring 170, 5 new
HQ: San Francisco, United States • LLM & Generative AI • AI Evaluation & Observability • AI Training Data & Annotation • 1001-5000 employees • Est. 2016 • Trust B • Hiring 170, 5 new
Scale AI is an American company founded in San Francisco in 2016 that supplies the training data, evaluation and tooling behind large artificial intelligence systems. It began with human-labelled annotation for autonomous driving and computer vision, then expanded into reinforcement learning from human feedback, expert data generation, model evaluation and full-stack deployment platforms for enterprises and governments. In 2025 Meta acquired a large minority stake and hired co-founder Alexandr Wang, after which the company continued under new leadership serving defence, public sector and commercial customers.
Jason Droege · Chief Executive Officer · decision-maker
Lucy Guo · Co-Founder · founder
Alexandr Wang · Co-Founder & former Chief Executive Officer · founder
No careers mailbox known
Likely inactiveTop 25% pay 1 year ago Long shot · 8%

Tech Lead Manager- MLRE, ML Systems

$290k – $363k per year • Equity • Lead • In office (San Francisco, New York, United States) • Full-Time • Staff
Flash Attention
CUDA Toolkit
RLHF
PyTorch
LLM
CUDA
PPO
GRPO
Post-training
Multimodal AI
Function Calling
Tool Use
+4
Apply
Likely inactiveTop 25% pay 1 year ago Long shot · 4%

ML Research Engineer, ML Systems

$190k – $237k per year • Equity • In office (San Francisco, Seattle, New York, United States) • Full-Time
Flash Attention
CUDA Toolkit
PyTorch
LLM
CUDA
RLHF
Multimodal AI
Function Calling
Post-training
Tool Use
+2
Apply

Tenstorrent

tenstorrent.com
tenstorrent.com • HQ: Toronto, Canada • Semiconductor IP & Chip Design Services • AI Chips & Accelerators • Processors, MCUs & AI Chips • 1001-5000 employees • Est. 2016 • Trust A • Hiring 113, 6 new
HQ: Toronto, Canada • Semiconductor IP & Chip Design Services • AI Chips & Accelerators • Processors, MCUs & AI Chips • 1001-5000 employees • Est. 2016 • Trust A • Hiring 113, 6 new
Tenstorrent is a Canadian semiconductor company founded in 2016 that designs artificial intelligence accelerators and RISC-V processors, and it is unusual in the field for licensing its designs as intellectual property as well as selling hardware. It is led by Jim Keller, the chip architect behind AMD's Zen and Apple's early mobile processors, and its argument against Nvidia is openness: an open instruction set, open source software stack and licensable designs rather than a closed platform. Headquartered in Toronto and backed by investors including Samsung, Hyundai and Bezos Expeditions, it competes in a market where software maturity matters as much as silicon.
Ljubisa Bajic · Founder · decision-maker
Alexander Munn · Executive Officer · decision-maker
Keith Witek · Director · decision-maker
James Keller · Executive Officer, Director · decision-maker
Careers mailbox is known
Open 384 daysVerified live · 2 hours ago 1 year ago Long shot · 4%

Machine Learning Engineer, AI Models

In office (Cyprus) • Full-Time
C++
C++
TensorFlow C++
PyTorch C++
AI/ML
Flash Attention
CUDA Toolkit
Quantization
TensorFlow
PyTorch
CUDA
Edge AI
+2
Apply

Anthropic

anthropic.com
anthropic.com • HQ: San Francisco, United States • Code Intelligence • AI Research Labs • AI Safety & Alignment • Foundation Models • LLM & Generative AI • 1001-5000 employees • Est. 2021 • Trust B • Hiring 646, 36 new
HQ: San Francisco, United States • Code Intelligence • AI Research Labs • AI Safety & Alignment • Foundation Models • LLM & Generative AI • 1001-5000 employees • Est. 2021 • Trust B • Hiring 646, 36 new
Anthropic is an American artificial intelligence safety and research company founded in 2021 by former OpenAI researchers, among them the siblings Dario and Daniela Amodei. It develops the Claude family of large language models and ships them through a consumer assistant, an enterprise developer platform and the Claude Code agentic coding tool, alongside open standards such as the Model Context Protocol. Incorporated as a public benefit corporation and headquartered in San Francisco, the company concentrates on interpretability, alignment and reliability research and counts Google and Amazon among its largest investors.
Jared Kaplan · Co-Founder & Chief Science Officer · decision-maker
Daniela Amodei · Co-Founder & President · decision-maker
Dario Amodei · Co-Founder & Chief Executive Officer · decision-maker
Tom Brown · Co-Founder · founder
No careers mailbox known
Open 388 daysTop 25% pay 1 year ago Long shot · 2%

Performance Engineer, GPU

$280k per year • Hybrid (San Francisco, United States) • Bachelor's Degree
Claude
Flash Attention
CUDA Toolkit
Quantization
JAX
Multimodal AI
PyTorch
CUDA
Triton
Anthropic
NCCL
NVLink
XLA
CUTLASS
Interpretability
+7
Apply

Baseten

baseten.co
baseten.co • HQ: San Francisco, United States • MLOps • AI Compute & Inference • 1001-5000 employees • Est. 2019 • Trust B • Hiring 103, 3 new
HQ: San Francisco, United States • MLOps • AI Compute & Inference • 1001-5000 employees • Est. 2019 • Trust B • Hiring 103, 3 new
Baseten is an American company founded in 2019 that runs machine learning models in production for companies that would rather not operate GPU infrastructure themselves. Its position is inference rather than training: it handles model packaging, autoscaling, cold start latency and multi-cloud capacity, which are the unglamorous problems that determine whether a model-powered product is fast and affordable enough to ship. Headquartered in San Francisco and backed at a multi-billion dollar valuation, it serves companies deploying open and custom models, and it competes with both the hyperscalers and the model providers' own hosted endpoints.
Sarah Guo · Executive Officer, Director · decision-maker
Tuhin Srivastava · Executive Officer, Director · decision-maker
Danielle Herzberg · Executive Officer · decision-maker
Amir Haghighat · Executive Officer, Director · decision-maker
No careers mailbox known
Open 455 daysTop 25% pay 1 year ago Long shot · 3%

Software Engineer - GPU Kernels

$180k – $360k per year • Hybrid (San Francisco, New York, Seattle, United States, Toronto, Montreal, Canada) • Full-Time
C++
AI/ML
Cursor
CUDA Toolkit
Embeddings
Quantization
Mixture of Experts
CUDA
Baseten
Machine Learning
Flash Attention
Transformers
Triton
CUTLASS
DevOps
AWS
Management
Notion
+7
Apply
Machines can read this selection · checked 20:17 UTC
Flash Attention Jobs - Remote & On-site
Frequently asked questions
How many Flash Attention jobs are available on Alion?
There are currently 15 active listings from 11 hiring companies on Alion, refreshed daily.
What salary do Flash Attention jobs pay?
The average salary is around $342,083 per year based on the current listings.
Are remote options available for Flash Attention jobs?
Yes, many of these roles offer remote or hybrid work, and you can filter listings by remote, relocation and visa sponsorship.
How do I apply for Flash Attention jobs?
Open any listing on Alion and apply directly with the employer through the provided link - no lengthy sign-up required.