[WIP] Add docs and device kwarg to to_nf4

[cpuhrsch]

Attempt to address #268

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/324

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❌ 3 New Failures

As of commit c89882a with merge base 729fa4d ():

NEW FAILURES - The following jobs have failed:

• Run Regression Tests / test (CUDA 2.2.2, linux.g5.12xlarge.nvidia.gpu, torch==2.2.2, cuda, 12.1) / linux-job (gh)
  test/dtypes/test_nf4.py::TestFSDPOps::test_to_nf4_cuda
• Run Regression Tests / test (CUDA 2.3, linux.g5.12xlarge.nvidia.gpu, torch==2.3.0, cuda, 12.1) / linux-job (gh)
  test/dtypes/test_nf4.py::TestFSDPOps::test_to_nf4_cuda
• Run Regression Tests / test (CUDA Nightly, linux.g5.12xlarge.nvidia.gpu, --pre torch==2.4.0.dev20240603+cu121 --index-ur... / linux-job (gh)
  test/dtypes/test_nf4.py::TestFSDPOps::test_to_nf4_cuda

This comment was automatically generated by Dr. CI and updates every 15 minutes.