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
New features:
jax.P which is an alias for jax.sharding.PartitionSpec.jax.tree.reduce_associative.Breaking changes:
jax.stages.OutInfo has been replaced with .jax.ShapeDtypeStructjax.jit now requires fun to be passed by position, and additional
arguments to be passed by keyword. Doing otherwise will result in an error
starting in v0.7.x. This raised a DeprecationWarning in v0.6.x.Layout, .layout, .input_layouts and .output_layouts have been
renamed to Format, .format, .input_formats and .output_formatsDeviceLocalLayout, .device_local_layout have been renamed to Layout
and .layoutjax.experimental.shard module has been deleted and all the APIs have been
moved to the jax.sharding endpoint. So use jax.sharding.reshard,
jax.sharding.auto_axes and jax.sharding.explicit_axes instead of their
experimental endpoints.lax.infeed and lax.outfeed were removed, after being deprecated in
JAX 0.6. The transfer_to_infeed and transfer_from_outfeed methods were
also removed the Device objects.jax.extend.core.primitives.pjit_p primitive has been renamed to
jit_p, and its name attribute has changed from "pjit" to "jit".
This affects the string representations of jaxprs. The same primitive is no
longer exported from the jax.experimental.pjit module.jax.extend.backend.add_clear_backends_callback
has been removed. Users should use jax.extend.backend.register_backend_cache
instead.Deprecations:
jax.dlpack.SUPPORTED_DTYPES is deprecated; please use the new
jax.dlpack.is_supported_dtype function.jax.scipy.special.sph_harm has been deprecated following a similar
deprecation in SciPy; use jax.scipy.special.sph_harm_y instead.jax.interpreters.xla, the previously deprecated symbols
abstractify and pytype_aval_mappings have been removed.jax.interpreters.xla.canonicalize_dtype is deprecated. For
canonicalizing dtypes, prefer jax.dtypes.canonicalize_dtype.
For checking whether an object is a valid jax input, prefer
jax.core.valid_jaxtype.jax.core, the previously deprecated symbols AxisName,
ConcretizationTypeError, axis_frame, call_p, closed_call_p,
get_type, trace_state_clean, typematch, and typecheck have been
removed.jax.lib.xla_client, the previously deprecated symbols
DeviceAssignment, get_topology_for_devices, and mlir_api_version
have been removed.jax.extend.ffi was removed after being deprecated in v0.5.0.
Use {mod}jax.ffi instead.jax.lib.xla_bridge.get_compile_options is deprecated, and replaced by
jax.extend.backend.get_compile_options.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.
A feature-rich command-line audio/video downloader
:octocat: 分享 GitHub 上有趣、入门级的开源项目。Share interesting, entry-level open source projects on GitHub.
Python tool for converting files and office documents to Markdown.