Spazio dei nomi/nome del pacchetto: jax.interpreters.xla
Metodo/funzione: axis_groups
Esempi su hotexamples.com: 2
axis_groups in Python: 2 esempi trovati. Questi sono i migliori esempi reali in Python per jax.interpreters.xla.axis_groups, estratti da progetti open source. Li puoi valutare, per aiutarci a migliorare la qualità dei nostri esempi.