Google JAX
id:
google-jax-181-10563746
title:
Google JAX
text:
Google JAX is a machine learning framework for transforming numerical functions. It is described as bringing together a modified version of autograd and TensorFlow's XLA. It is designed to follow the structure and workflow of NumPy as closely as possible and works with various existing frameworks such as TensorFlow and PyTorch. The primary functions of JAX are:
- grad: automatic differentiation
- jit: compilation
- vmap: auto-vectorization
- pmap: SPMD programming
brand slug:
wiki
category slug:
encyclopedia
description:
Machine Learning framework designed for parallelization and autograd.
original url:
https://en.wikipedia.org/wiki/Google_JAX
date created:
2022-06-18T20:56:18Z
date modified:
2024-09-06T00:58:42Z
main entity:
{"identifier":"Q107382149","url":"https://www.wikidata.org/entity/Q107382149"}
image:
{"content_url":"https://upload.wikimedia.org/wikipedia/commons/8/86/Google_JAX_logo.svg","width":512,"height":296}
fields total:
13
integrity:
16