Update __init__.py

luyug · Feb 18, 2024 · 0c33638 · 0c33638
1 parent e4234c9
commit 0c33638
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/grad_cache/cachex/__init__.py b/src/grad_cache/cachex/__init__.py
@@ -1,3 +1,3 @@
-from .functional import chunk_encode, cache_grad, unchunk_args
+from .functional import chunk_encode, cache_grad, unchunk_args, grad_cached
 from .tree_utils import tree_chunk, tree_unchunk