We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d33cda0 commit 5940630Copy full SHA for 5940630
pytensor/link/mlx/dispatch/core.py
@@ -1,16 +1,3 @@
1
-"""
2
-pytensor/link/mlx/dispatch/basic.py
3
------------------------------------
4
-
5
-First-cut MLX translations for the most common tensor Ops.
6
7
-The structure intentionally follows pytensor's JAX dispatcher so that
8
-once these kernels stabilise they can be optimised further (e.g. fusing
9
-element-wise graphs, adding in-place updates, RNG thinning, etc.).
10
11
12
-from __future__ import annotations
13
14
import mlx.core as mx
15
import numpy as np
16
0 commit comments