add the float8 NUZ types to torchao/prototype/mx_formats/fp_format_spec.py

Open Beginner friendly
#365 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Assessment

Difficulty
2/5
Estimated time
1-3 hours
Newbie friendliness
62/100
Issue type
Feature
Clarity
Mostly clear
Activity status
Stale
Tech stack
python, pytorch

Research direction

Start in torchao/prototype/mx_formats/fp_format_spec.py and compare the existing format specifications with the low-precision dtypes supported in PyTorch. Add the float8 NUZ types and verify that the script covers the intended supported dtypes.

Written by the indexing model from the issue text.

Description

We should cover all the low precision dtypes supported in PT in this script, I wish I had this today as I was working on adding some logic for the NUZ types

Dominant language
Python
Stars
3k
Forks
627
Avg merge
4d 14h
Merged PRs (30d)
34

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

More from pytorch/ao

All issues in pytorch/ao

Similar issues

More Python issues

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.