JIT: частично или со статическими аргнумами? НЕ ХАЗБЕБА ⇐ Python

[Anonymous]

Я немного потерян в том, что именно происходит и какой вариант выбрать. Давайте перейдем к примеру: < /p>

Код: Выделить всё

import jax
from functools import partial
from typing import List

def dummy(a: int, b: List[str]):
return a + 1

как b аргумент изменен, Jittive со статическими именами Argname не удастся:

Код: Выделить всё

j_dummy = jax.jit(dummy, static_argnames=['b'])
j_dummy(2, ['kek'])
ValueError: Non-hashable static arguments are not supported

Однако, если мы делаем частичный : jp_dummy = jax.jit (частичный (Dummy, b = ['kek'])) , мы стремимся к цели. Каким -то образом, частичный объект действительно имеет метод __hash __ , поэтому мы можем проверить его с помощью хэша (частичный (манекен, b = ['kek'])) .
Итак, я немного потерян здесь: как я должен продолжить в более широкой картине? Должен ли я создавать частичные функции с любыми аргументами, а затем их считать или я должен попытаться сохранить свои аргументы с хранением? Каковы ситуации, когда один подход лучше других? Есть ли недостатки?

Подробнее здесь: https://stackoverflow.com/questions/791 ... le-partial