huggingface / huggingface/aokit

Roadmap: unified `aokit.compile` API and multi-architecture support

Open
#10 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
3
Forks
3
Avg merge
3m
Merged PRs (30d)
1

Description

Tracking two major features currently planned for AOKit:

**1. Unified `aokit.compile` API**
A single entry point covering the full workflow:
- Example capture
- Automatic dynamic shape inference
- Export
- Compilation

**2. Multi-architecture support**
- GPU-side: multiple SM architectures
- CPU-side: aarch64 (increasingly common)

Feedback on relevance and priority is very welcome.

---
Requested by Sayak Paul (Slack U03AU4E7DJB) via Moon Bot _(no GitHub handle mapped)_.
Slack thread: https://huggingface.slack.com/archives/C013QGNQ3T4/p1781694239578039?thread_ts=1781694239.578039&cid=C013QGNQ3T4
Agent trace: https://huggingface.co/buckets/huggingface/moon-bot-memory/tree/sessions/2026-06-18T10-09-48-525Z_0467c3f7-2733-4e1b-93d6-ad322f3f58b2.jsonl

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.