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
Đang mở
- Ngôn ngữ chính
- Julia
- Star
- 1.1k
- Fork
- 86
- Merge trung bình
- 1 ngày 22 giờ
- Pull request đã merge (30 ngày)
- 3
Mô tả
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
Hướng dẫn đóng góp
Chưa lập chỉ mục được hướng dẫn đóng góp cho kho mã nguồn này
Đánh giá
Issue này chưa được đánh giá.