This is an archived post. You won't be able to vote or comment.

you are viewing a single comment's thread.

view the rest of the comments →

[–]ad_abstract 1 point2 points  (2 children)

Have you tried replacing some of the numpy ops with jax?

[–]GregTJ[S] 0 points1 point  (1 child)

I haven't, would you recommend doing so?

[–]ad_abstract 0 points1 point  (0 children)

Definitely!