diff --git a/tinygrad/nn/__init__.py b/tinygrad/nn/__init__.py index 3039e0a8a1..edb72684e4 100644 --- a/tinygrad/nn/__init__.py +++ b/tinygrad/nn/__init__.py @@ -2,7 +2,6 @@ from __future__ import annotations import math from tinygrad.tensor import Tensor from tinygrad.dtype import dtypes -from tinygrad.device import is_dtype_supported as is_dtype_supported from tinygrad.helpers import prod, make_tuple, flatten, USE_ATOMICS from tinygrad.nn import optim, state, datasets # noqa: F401