Assistance with Optimizing JAX-Haiku Application #15003
Unanswered
timhdesilva
asked this question in
Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Hi!
I've written a package using JAX and Haiku for use on an academic project, but am still relatively new to both paradigms. I am wondering if anyone could point me to the best way to find someone who's an expert that would be able to assist me in assessing whether it is possible to optimize the structure of the code further to run faster on an Nvidia GPU. I have the resources to pay for this, but am not sure if this is the best place to be posting this request. I've had trouble finding freelance workers on the usual platforms (e.g., Upwork, Codementor) who are familiar enough with JAX and Haiku!
Any thoughts would be much appreciated. And of course if anyone is interested, please comment below. For reference, the package located here (https://github.com/marcdelabarrera/nndp). There is an example application that has already been written in the examples/ folder that runs with no issue. I would like to see if there is any way to optimize this application further for running on an Nvidia GPU.
Thank you very much and apologies if this is the wrong place to be posting this!
Beta Was this translation helpful? Give feedback.
All reactions