mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-04-29 03:00:14 -04:00
e3fa9896b7
* start function and add walk rewrite * work * add function on feed_forward * llm progress * stuff * none of that
13 lines
674 B
Python
13 lines
674 B
Python
import os
|
|
if int(os.getenv("TYPED", "0")):
|
|
from typeguard import install_import_hook
|
|
install_import_hook(__name__)
|
|
from tinygrad.tensor import Tensor # noqa: F401
|
|
from tinygrad.engine.jit import TinyJit # noqa: F401
|
|
from tinygrad.function import function # noqa: F401
|
|
from tinygrad.uop.ops import UOp
|
|
Variable = UOp.variable
|
|
from tinygrad.dtype import dtypes # noqa: F401
|
|
from tinygrad.helpers import GlobalCounters, fetch, Context, getenv # noqa: F401
|
|
from tinygrad.device import Device # noqa: F401
|