search

pytorch-labs / float8_experimental

This repository contains the experimental PyTorch native float8 training UX
BSD 3-Clause "New" or "Revised" License
194 stars 18 forks source link

Add sanity checks to dtensor tests #302

Closed drisspg closed 2 weeks ago

drisspg commented 2 weeks ago

Stack from ghstack (oldest at bottom):

drisspg commented 2 weeks ago

Wow I was being so dumb..