Write faster numerical computing code with JAX.
What it does
Guides development of efficient numerical computing and machine-learning code with JAX.
When to use it
Use when working on high-performance numerical computing or machine learning with JAX.
How to use it
It changes how the agent writes JAX code by emphasizing pure functions, immutability, transformations, and JIT optimization.
What you provide
- An existing project
Latest review
What you need · 1
The coding guidance is for JAX, including its arrays and transformations.
Access · 0
No additional actions listed.