AI & Machine Learning

JAX Skills

High-performance numerical computing and machine learning workflows using JAX. Supports array operations, automatic differentiation, JIT compilation, RNN-style scans, map/reduce operations, and gradient computations. Ideal for scientific computing, ML models, and dynamic array transformations.

Publisher / source
benchflow-ai/skillsbench (found on GitHub top repositories)
Upstream record
benchflow-ai/skillsbench
Version
Version 1
Published
Publication review
Approved for publication · recorded passed scan 2026-11-07
Content fingerprint
2cff1fe8e449fe60ab8009980c4e44eaf0d6bc7ca14ae03fc1e5c7f7e82eb6d1