mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-14 10:56:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix typo in jax.typing module doc

Browse Source
This commit is contained in:
Luke Yang 2024-09-11 23:34:03 +10:00 committed by GitHub
parent 49dd6ed8d8
commit e5107c125d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/typing.py
View File

@ -24,7 +24,7 @@ The currently-available types are:
- :obj:`jax.typing.ArrayLike`: annotation for any value that is safe to implicitly cast to
a JAX array; this includes :class:`jax.Array`, :class:`numpy.ndarray`, as well as Python
builtin numeric values (e.g. :class:`int`, :class:`float`, etc.) and numpy scalar values
(e.g. :class:`numpy.int32`, :class:`numpy.flota64`, etc.)
(e.g. :class:`numpy.int32`, :class:`numpy.float64`, etc.)
- :obj:`jax.typing.DTypeLike`: annotation for any value that can be cast to a JAX-compatible
dtype; this includes strings (e.g. `'float32'`, `'int32'`), scalar types (e.g. `float`,
`np.float32`), dtypes (e.g. `np.dtype('float32')`), or objects with a dtype attribute