Unclaimed project
Are you a maintainer of jax? Claim this project to take control of your public changelog and roadmap.
Changelog
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
AutoGPT is the vision of accessible AI for everyone, to use and to build on. Our mission is to provide the tools, so that you can focus on what matters.
Stable Diffusion web UI
🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
A feature-rich command-line audio/video downloader
JAX v0.9.0.1 is identical to v0.9.0 with the commits from the following four PRs patched in:
JAX v0.8.3 is identical to v0.8.2 with the following two bug fixes patched in:
New features:
jax.thread_guard, a context manager that detects when devices
are used by multiple threads in multi-controller JAX.Bug fixes:
magma_zgeqp3_gpu)
in MAGMA 2.9.0 when using use_magma=True and pivoting=True.
(#34145).Deprecations:
jax.lax.pvary has been deprecated.
Please use jax.lax.pcast(..., to='varying') as the replacement.jax.numpy.arange now result in a
deprecation warning, because the output is poorly-defined.jax.core a number of symbols are newly deprecated including:
call_impl, get_aval, mapped_aval, subjaxprs, `set_curr...New features:
jax.jit now supports the decorator factory pattern; i.e instead of
writing
@functools.partial(jax.jit, static_argnames=['n'])
def f(x, n):
...
@jax.jit(static_argnames=['n'])
def f(x, n):
...
Changes:
jax.lax.linalg.eigh now accepts an implementation ar...Breaking changes:
jax.pmap implementation to one implemented in
terms of jax.jit and jax.shard_map. jax.pmap is in maintenance mode
and we encourage all new code to use jax.shard_map directly. See the
migration guide for
more information.auto= parameter of `jax.experim...Breaking changes:
jax.dlpack.from_dlpack no longer accepts a DLPack capsule. This
behavior was deprecated and is now removed. The function must be called
with an array implementing __dlpack__ and __dlpack_device__.Changes
New features
Changes
jax.set_mesh which acts as a global setter and a context manager.
Removed jax.sharding.use_mesh in favor of jax.set_mesh.New features:
jax.P which is an alias for jax.sharding.PartitionSpec.jax.tree.reduce_associative.Breaking changes:
New features:
jax.tree.broadcast which implements a pytree prefix broadcasting helper.Changes