Agent: Cursor, Claude CodeLLM: GPT-4, Claude 3.5#machine-learning#numerical-computing#automatic-differentiation#gpu-acceleration#deep-learning
JAX is a Python library for accelerator-oriented array computation and program transformation, designed for high-performance numerical computing and large-scale machine learning. It provides automatic differentiation, JIT compilation, and vectorization capabilities.