Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add recipe: custom ops with applications in AI models #14

Merged
merged 1 commit into from
Mar 4, 2025

Conversation

BradLarson
Copy link
Collaborator

Adds a recipe for the more advanced custom ops examples: top-K token sampling and a fused attention layer.

Marking this as a draft for now, because significant additions need to be made to the README to fill in relevant information about the two operations and what people should learn as they examine the code. The rest of the scaffolding is there, as well as all example code and what's needed to run them.

@BradLarson BradLarson marked this pull request as ready for review March 4, 2025 15:52
@BradLarson
Copy link
Collaborator Author

Even though the wording is just a very, very quick overview here and will be expanded upon, I'm going to merge along with the other examples so that others can add to this without needing to work off of my branch.

@BradLarson BradLarson merged commit 1387c37 into modular:main Mar 4, 2025
1 check passed
@BradLarson BradLarson deleted the custom-ops-ai-applications branch March 6, 2025 15:09
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant