Instrument

JAX

Level: 7/10 | Type: High-Performance Numerical Computing & ML Framework | Paradigm: Composable Function Transformations, Autograd & XLA Compilation

7/10 level — composable function transformations (grad, jit, vmap), XLA GPU compilation, functional pure-state ML modeling, and numerical research.

Level: 7/10

JAXXLAFunctional MLAutogradVectorization

Competent experience in Google JAX (7/10). Utilizing functional transformations including automatic differentiation (grad), just-in-time compilation to XLA (jit), and vectorization batching (vmap) for accelerated research simulations and numerical computing.

Where this has actually been used

Projects

NEED — Neuro-Evolution for Effective Decision

Loading content…