jax.experimental

E438360

jax.experimental is a submodule of the JAX library that provides access to experimental, unstable, or cutting-edge numerical and machine learning features not yet part of the stable API.

All labels observed (7)

How this entity was disambiguated

Statements (35)

Predicate Object
instanceOf Python module ⓘ
software component ⓘ
apiStatus experimental ⓘ
documentationWarning APIs may be removed or changed ⓘ
domain machine learning ⓘ
numerical computing ⓘ
ecosystem Python scientific computing ⓘ
hasPurpose to expose features not yet in the stable JAX API ⓘ
isNamespaceFor jax.experimental.callback ⓘ
jax.experimental.checkify ⓘ
linked to: jax.experimental

jax.experimental.compilation_cache ⓘ
linked to: jax.experimental

jax.experimental.global_device_array ⓘ
jax.experimental.host_callback ⓘ
jax.experimental.jax2tf ⓘ
jax.experimental.loops ⓘ
linked to: jax.experimental

jax.experimental.maps ⓘ
jax.experimental.multihost_utils ⓘ
jax.experimental.optimizers ⓘ
linked to: jax.experimental

jax.experimental.pjit ⓘ
jax.experimental.shard_map ⓘ
jax.experimental.sparse ⓘ
jax.experimental.stax ⓘ
library JAX ⓘ
mayChange without backward compatibility guarantees ⓘ
partOf jax ⓘ
linked to: jax.experimental
programmingLanguage Python ⓘ
provides cutting-edge machine learning functionality ⓘ
cutting-edge numerical functionality ⓘ
experimental features ⓘ
unstable APIs ⓘ
requires jax ⓘ
linked to: jax.experimental
stability unstable ⓘ
targetUsers advanced JAX users ⓘ
early adopters of new JAX features ⓘ
researchers ⓘ

How these facts were elicited

Referenced by (9)

Full triples — surface form annotated when it differs from this entity's canonical label.

JAX → hasComponent → jax.experimental ⓘ
Flax → compatibleWith → NumPy API via JAX ⓘ
linked to: jax.experimental
jax.random → partOf → jax ⓘ
linked to: jax.experimental
jax.experimental → partOf → jax ⓘ
linked to: jax.experimental
jax.experimental → requires → jax ⓘ
linked to: jax.experimental
jax.experimental → isNamespaceFor → jax.experimental.checkify ⓘ
linked to: jax.experimental
jax.experimental → isNamespaceFor → jax.experimental.loops ⓘ
linked to: jax.experimental
jax.experimental → isNamespaceFor → jax.experimental.compilation_cache ⓘ
linked to: jax.experimental
jax.experimental → isNamespaceFor → jax.experimental.optimizers ⓘ
linked to: jax.experimental