Sketched Linear Layers
======================

Backward time for linear on NVIDIA T4 with Tensor Core implementation

.. image:: ../_static/63.png
   :alt: Backward pass time (ms) for SKLinear vs nn.Linear on NVIDIA T4 (Tensor Cores), varying layer width
   :width: 600px
   :align: center

Backward time for linear on NVIDIA P100 without Tensor Cores

.. image:: ../_static/64.png
   :alt: Backward pass time (ms) for SKLinear vs nn.Linear on NVIDIA P100 (no Tensor Cores), varying layer width
   :width: 600px
   :align: center

Forward time for linear on NVIDIA T4 with Tensor Core implementation

.. image:: ../_static/65.png
   :alt: Forward pass time (ms) for SKLinear vs nn.Linear on NVIDIA T4 (Tensor Cores), varying layer width
   :width: 600px
   :align: center

Forward time for linear on NVIDIA P100 without Tensor Cores

.. image:: ../_static/66.png
   :alt: Forward pass time (ms) for SKLinear vs nn.Linear on NVIDIA P100 (no Tensor Cores), varying layer width
   :width: 600px
   :align: center