jax-ml/jax

lax.conv_transpose has different convention than TF, document it and make switching easy?

Aperta

#14.337 aperta il 7 feb 2023

Vedi su GitHubGuida contributor
 (11 commenti) (0 reazioni) (0 assegnatari)Python (3738 fork)github user discovery
documentationenhancementgood first issue

Metriche repository

Star
 (36.174 stelle)
Metriche merge PR
 (Merge medio 2g 19h) (439 PR mergiate in 30 g)

Descrizione

See screenshot:

Though the example uses Haiku, Haiku is just calling through to lax.conv_transpose.

The difference is in reversing the spatial axes of the kernel.

We can do at least two things to make this less surprising and more easily fixable:

  1. document the difference, and
  2. add an argument to lax.conv_transpose to instead use the TF convention.

Guida contributor

Direzione di ricerca
Ispeziona l'implementazione di 'lax.conv transpose' in JAX, confronta la convenzione degli assi del kernel con TensorFlow, quindi aggiungi un parametro per cambiare convenzione e aggiorna la documentazione.
Tech stack
python
Dominio
backend
Tipo issue
Funzionalità
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
2
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
Mezza giornata
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
PythonJAX
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
65