Voici la liste des méthodes et propriétés proposé par la bibliothèque JAX pour Python :
| Nom | Description |
|---|---|
| jax.Array.addressable_shards | Cette propriété retourne la liste des shards du tableau qui sont directement accessibles depuis Python, utile pour inspecter ou manipuler les données distribuées. |
| jax.Array.all | Cette méthode retourne True si tous les éléments du tableau sont évalués comme True, sinon False. |
| jax.Array.any | Cette méthode retourne True si au moins un élément du tableau est évalué comme True, sinon False. |
| jax.Array.argmax | Cette méthode retourne l'indice de l'élément maximum du tableau le long d'un axe donné. |
| jax.Array.argmin | Cette méthode retourne l'indice de l'élément minimum du tableau le long d'un axe donné. |
| jax.Array.argpartition | Cette méthode retourne les indices qui partitionnent le tableau selon l'ordre spécifié, utile pour des sélections partielles rapides. |
| jax.Array.argsort | Cette méthode retourne les indices qui trieraient le tableau le long d'un axe donné. |
| jax.Array.astype | Cette méthode convertit les éléments du tableau dans un autre type de données (dtype). |
| jax.Array.at | Cette propriété/méthode permet d'effectuer des opérations immuables sur des indices spécifiques, comme des mises à jour ou des ajouts. |
| jax.Array.choose | Cette méthode sélectionne des éléments selon un tableau d'indices ou un choix donné, similaire à numpy.choose. |
| jax.Array.clip | Cette méthode limite les valeurs du tableau entre des bornes a_min et a_max. |
| jax.Array.compress | Cette méthode sélectionne les éléments du tableau en fonction d'un masque booléen. |
| jax.Array.committed | Cette propriété indique si le tableau est committed sur le dispositif, c'est-à-dire pleinement disponible en mémoire sur CPU/GPU/TPU. |
| jax.Array.conj | Cette méthode retourne le conjugué complexe du tableau. |
| jax.Array.conjugate | Cette méthode est un alias de conj et retourne le conjugué complexe du tableau. |
| jax.Array.copy | Cette méthode crée une copie du tableau, souvent utilisée pour éviter les effets de bord. |
| jax.Array.copy_to_host_async | Cette méthode copie le tableau de manière asynchrone sur l'hôte (CPU), utile pour récupérer des données d'un GPU ou TPU sans bloquer le programme. |
| jax.Array.cumprod | Cette méthode retourne le produit cumulatif des éléments du tableau le long d'un axe spécifié. |
| jax.Array.cumsum | Cette méthode retourne la somme cumulative des éléments du tableau le long d'un axe spécifié. |
| jax.Array.device | Cette propriété retourne le dispositif (CPU, GPU ou TPU) sur lequel le tableau est stocké. |
| jax.Array.diagonal | Cette méthode retourne la diagonale du tableau ou de ses matrices, avec options pour axes et offset. |
| jax.Array.dot | Cette méthode effectue le produit matriciel ou scalaire du tableau avec un autre tableau compatible. |
| jax.Array.dtype | Cette propriété retourne le type de données (dtype) des éléments du tableau, par exemple float32 ou int64. |
| jax.Array.flat | Cette propriété retourne une vue aplatie du tableau, permettant d'itérer sur tous les éléments comme sur un vecteur 1D. |
| jax.Array.flatten | Cette méthode retourne une copie aplatie du tableau, transformant toutes ses dimensions en un seul vecteur 1D. |
| jax.Array.global_shards | Cette propriété retourne la liste de tous les shards du tableau, y compris ceux qui ne sont pas directement accessibles depuis Python. |
| jax.Array.imag | Cette propriété retourne la partie imaginaire des éléments du tableau. |
| jax.Array.is_fully_addressable | Cette propriété retourne True si tous les shards du tableau sont accessibles depuis Python, sinon False. |
| jax.Array.is_fully_replicated | Cette propriété retourne True si tous les shards du tableau sont identiques (répliqués) sur tous les dispositifs, sinon False. |
| jax.Array.item | Cette méthode retourne un élément du tableau sous forme de scalaire Python, en utilisant éventuellement un index. |
| jax.Array.itemsize | Cette propriété retourne la taille en octets d'un élément du tableau. |
| jax.Array.max | Cette méthode retourne la valeur maximale des éléments du tableau, éventuellement le long d'un axe donné. |
| jax.Array.mean | Cette méthode retourne la moyenne des éléments du tableau, éventuellement le long d'un axe donné. |
| jax.Array.min | Cette méthode retourne la valeur minimale des éléments du tableau, éventuellement le long d'un axe donné. |
| jax.Array.nbytes | Cette propriété retourne le nombre total d'octets utilisés par le tableau. |
| jax.Array.ndim | Cette propriété retourne le nombre de dimensions du tableau. |
| jax.Array.nonzero | Cette méthode retourne les indices des éléments non nuls du tableau sous forme de tuple de tableaux. |
| jax.Array.prod | Cette méthode retourne le produit de tous les éléments du tableau, éventuellement le long d'un axe donné. |
| jax.Array.ptp | Cette méthode retourne la plage (max - min) des valeurs du tableau, éventuellement le long d'un axe donné. |
| jax.Array.ravel | Cette méthode retourne une vue aplatie 1D du tableau sans copier les données. |
| jax.Array.real | Cette propriété retourne la partie réelle des éléments du tableau. |
| jax.Array.repeat | Cette méthode réplique les éléments du tableau selon un nombre donné de répétitions le long d'un axe spécifié. |
| jax.Array.reshape | Cette méthode retourne une nouvelle vue du tableau avec la forme spécifiée. |
| jax.Array.round | Cette méthode arrondit les éléments du tableau au nombre de décimales spécifié. |
| jax.Array.searchsorted | Cette méthode retourne les indices où insérer des valeurs pour conserver l'ordre trié du tableau. |
| jax.Array.shape | Cette propriété retourne la forme du tableau sous forme de tuple d'entiers. |
| jax.Array.sharding | Cette propriété retourne les informations sur la répartition (sharding) du tableau sur plusieurs dispositifs. |
| jax.Array.size | Cette propriété retourne le nombre total d'éléments du tableau. |
| jax.Array.sort | Cette méthode retourne le tableau trié le long d'un axe spécifié. |
| jax.Array.squeeze | Cette méthode retourne le tableau sans dimensions unitaires, supprimant les axes de longueur 1. |
| jax.Array.std | Cette méthode retourne l'écart type des éléments du tableau, éventuellement le long d'un axe donné. |
| jax.Array.sum | Cette méthode retourne la somme des éléments du tableau, éventuellement le long d'un axe donné. |
| jax.Array.swapaxes | Cette méthode retourne le tableau avec deux axes échangés. |
| jax.Array.take | Cette méthode retourne les éléments sélectionnés par indices, le long d'un axe donné. |
| jax.Array.to_device | Cette méthode copie le tableau sur un dispositif spécifique (CPU, GPU ou TPU). |
| jax.Array.trace | Cette méthode retourne la somme de la diagonale du tableau ou d'une matrice. |
| jax.Array.transpose | Cette méthode retourne le tableau avec les axes réordonnés selon les paramètres donnés. |
| jax.Array.var | Cette méthode retourne la variance des éléments du tableau, éventuellement le long d'un axe donné. |
| jax.Array.view | Cette méthode retourne une vue du tableau sans copier les données, similaire à NumPy. |
| jax.Array.T | Cette propriété retourne la transposée du tableau. |
| jax.Array.mT | Cette propriété retourne la transposée matricielle (souvent identique à .T pour 2D). |
Dernière mise à jour : Lundi, le 2 février 2026