Op

Optimizing DeepSeek's NSA for TPUs – A Kernel Worklog

Hacker News

Optimizing DeepSeek's NSA for TPUs – A Kernel Worklog

I enjoyed reading DeepSeek's Natively Sparse Attention(NSA) and implemented a JAX version and Pallas Kernel(JAX's kernel language). There's a Google Colab link to follow along at the top of the blogpost. Enjoy!

Share card

Actual performance

2points
Did not reach leaderboard

Launch Intel predictions

Analyze your own launch →
Product HuntOn track for Day 1 leaderboard · Strong signals: google · Missing: mac, agents, macos
62%62% predicted probability of success on Product Hunt, based on ML models trained on real launch data.
best fitHighest predicted score across all platforms for this description.
Hacker NewsMay not resonate with HN audience · Strong signals: io · Missing: https docs, excited, just released
48%48% predicted probability of success on Hacker News, based on ML models trained on real launch data.
nativeThis product was originally launched on this platform.
BetaListMay not resonate with beta-testers · Missing: web3, chat, crypto
42%42% predicted probability of success on BetaList, based on ML models trained on real launch data.
Indie HackersIH features products with proven revenue · Missing: supports, reddit linkedin, podcasting
31%31% predicted probability of success on Indie Hackers, based on ML models trained on real launch data.
TrustMRRLess likely to generate early MRR · Strong signals: google · Missing: mobile apps, ios, personal
29%29% predicted probability of success on TrustMRR, based on ML models trained on real launch data.
AppSumoMay struggle as an AppSumo deal · Missing: plus, platform, intuitive
27%27% predicted probability of success on AppSumo, based on ML models trained on real launch data.
Acquire.comPre-revenue stage for this audience · Missing: arr, mrr, revenue
19%19% predicted probability of success on Acquire.com, based on ML models trained on real launch data.

Correct prediction on native model

Similar products

tokenmix
tokenmix39%Launch Intel prediction score: how likely this product is to succeed on its source platform, based on its name, tagline, and description.

#chatgpt #deepseek

Indie Hackers
DeepSeek PDF Exporter
DeepSeek PDF Exporter14%Launch Intel prediction score: how likely this product is to succeed on its source platform, based on its name, tagline, and description.

Export deepseek to pdf

Product Hunt+6
De
DeepSeek Your HN Profile47%Launch Intel prediction score: how likely this product is to succeed on its source platform, based on its name, tagline, and description.

DeepSeek Your HN Profile

Hacker News121
De
DeepSeek Janus Pro 7B API30%Launch Intel prediction score: how likely this product is to succeed on its source platform, based on its name, tagline, and description.

DeepSeek Janus Pro 7B API

Hacker News6
Sn
Snappy, local-first DeepSeek chat37%Launch Intel prediction score: how likely this product is to succeed on its source platform, based on its name, tagline, and description.

Snappy, local-first DeepSeek chat

Hacker News2
De
DeepSeek My User Agent32%Launch Intel prediction score: how likely this product is to succeed on its source platform, based on its name, tagline, and description.

DeepSeek My User Agent

Hacker News360
DeepSeek For Android
DeepSeek For Android40%Launch Intel prediction score: how likely this product is to succeed on its source platform, based on its name, tagline, and description.

DeepSeek's official AI assistant for free

Product Hunt+27
Awesome Deepseek Integration
Awesome Deepseek Integration46%Launch Intel prediction score: how likely this product is to succeed on its source platform, based on its name, tagline, and description.

DeepSeek API Integrations, Officially Recommended

Product Hunt+14
Locl.chat
Locl.chat42%Launch Intel prediction score: how likely this product is to succeed on its source platform, based on its name, tagline, and description.

The snappiest Deepseek chat

Product Hunt+12
AI Chat Finder (DeepSeek)
AI Chat Finder (DeepSeek)52%Launch Intel prediction score: how likely this product is to succeed on its source platform, based on its name, tagline, and description.

Chat Content Search for DeepSeek + More

Product Hunt+13