We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 50511fe commit 53e1663Copy full SHA for 53e1663
arraycontext/impl/jax/__init__.py
@@ -41,7 +41,7 @@
41
class EagerJAXArrayContext(ArrayContext):
42
"""
43
A :class:`ArrayContext` that uses
44
- :class:`jaxlib.xla_extension.DeviceArrayBase` instances for its base array
+ :class:`jax.Array` instances for its base array
45
class and performs all array operations eagerly. See
46
:class:`~arraycontext.PytatoJAXArrayContext` for a lazier version.
47
0 commit comments