Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ml-explore/mlx
/ unbatch_vjp_second
Function
unbatch_vjp_second
benchmarks/python/batch_matmul_bench.py:46–47 ·
view source on GitHub ↗
()
Source
from the content-addressed store, hash-verified
44
time_fn(unbatch_vjp_first)
45
46
def
unbatch_vjp_second():
47
return
mx.matmul(mx.transpose(a), c)
48
49
time_fn(unbatch_vjp_second)
50
Callers
nothing calls this directly
Calls
no outgoing calls
Tested by
no test coverage detected