9 lines
141 BLFS
Plaintext
9 lines
141 BLFS
Plaintext
name: jax-jit
|
|
description: JIT compile function
|
|
inputs:
|
|
fn: callable
|
|
args: any
|
|
outputs:
|
|
result: any
|
|
implementation: jax_skills.jit_run
|