Requirements for Outputs

General Guidelines

Arrays

• All arrays MUST be compatible with JAX (jnp.array) or convertible from Python lists.
• Use .npy, .npz, JSON, or pickle for saving arrays.

Operations

• Validate input types and shapes for all functions.
• Maintain numerical stability for all operations.
• Provide meaningful error messages for unsupported operations or invalid inputs.

JAX Skills

1. Loading and Saving Arrays