name: jax-save
description: Save JAX result to numpy file
inputs:
  data: any
  path: string
implementation: jax_skills.save