Yash Katariya e21aee18a8 Add deprecation warning for FROM_GDA usage since that argument is not required anymore. ...

PiperOrigin-RevId: 519781715

2023-03-27 11:33:11 -07:00

..

compilation_cache

Fix tests that were adding __annotations__ for some reason to CompileOptions.

2023-03-08 13:55:03 -08:00

gda_serialization

Remove references to jax.config.jax_array, which is always True at head.

2023-03-15 17:09:11 -07:00

jax2tf

[shape_poly] Fixed bug with dimension variables in unused args

2023-03-27 13:37:39 +02:00

sparse

Deprecated xla_call_p since it has been replaced with pjit.pjit_p

2023-03-23 11:44:42 -07:00

__init__.py

Change jax.ad, jax.xla, jax.pxla to point to the shims instead of the internal modules.

2023-02-09 13:31:40 -08:00

checkify.py

Document why 'import name as name' is used

2022-12-14 15:07:04 -08:00

custom_partitioning.py

Improve pytype inference for Sharding type.

2023-03-14 14:20:17 -07:00

host_callback.py

Deprecated xla_call_p since it has been replaced with pjit.pjit_p

2023-03-23 11:44:42 -07:00

jet.py

Fix mypy issue in jax/experimental/jet.py

2023-03-23 13:56:11 -07:00

maps.py

Remove _PositionalSemantics class since it is not used anymore because jax.Array always has GLOBAL semantics

2023-03-17 13:30:04 -07:00

mesh_utils.py

Enable more mesh shape assignment

2023-03-20 15:26:55 -07:00

multihost_utils.py

Add deprecation warning for FROM_GDA usage since that argument is not required anymore.

2023-03-27 11:33:11 -07:00

ode.py

Remove leading underscores in jax._src.numpy.util

2023-03-13 12:18:36 -07:00

pjit.py

Add deprecation warning for FROM_GDA usage since that argument is not required anymore.

2023-03-27 11:33:11 -07:00

rnn.py

Copy seq_lengths before creating descriptor

2023-03-27 10:59:44 -07:00

serialize_executable.py

Redefine compile_and_serialize as serialize(lowered.compile()).

2023-03-22 17:23:19 -07:00

shard_map.py

Deprecated xla_call_p since it has been replaced with pjit.pjit_p

2023-03-23 11:44:42 -07:00

x64_context.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00