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
Open
- Dominant language
- Julia
- Stars
- 1.1k
- Forks
- 86
- Avg merge
- 1d 22h
- Merged PRs (30d)
- 3
Description
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
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.