JuliaPy / JuliaPy/PythonCall.jl
PythonCall With Jax: Fast inference w/ numpy but does not work with jax.grad need jax.numpy, which is slow, for gradient
Aperta
- Lingua principale
- Julia
- Stelle
- 1.1k
- Fork
- 86
- Merge medio
- 1g 22h
- PR unite (30g)
- 3
Descrizione
I need Jax for something that Zygote cannot do well (meta-learning) and someone recommended PythonCall as a solution to some issues I was having with PyCall.
So far, PythonCall has been great. Things work and it is generally quite quick.
There is one pain point: jax.grad does not work with numpy.array
Guida per i contributori
Nessuna guida per i contributori indicizzata per questo repository
Valutazione
Questa issue non è ancora stata valutata.