JAX and Functional ML
JAX jit, grad, vmap, pmap - functional transformations for high-performance ML, XLA compilation, and NumPy-compatible ML research.
2 docs tagged with "jax"
View all tagsModule 01 - Scientific Python Stack Overview
NumPy, Pandas, SciPy, Matplotlib, scikit-learn, PyTorch, and JAX - the complete Python stack for AI/ML engineering.