nnetsauce with and without jax for GPU acceleration | R bloggers

nnetsauce with and without jax for GPU acceleration | R bloggers

[This article was first published on T. Moudiki’s Webpage – R, and kindly contributed to R-bloggers]. (You can report a problem with the content on this page here)


Want to share your content on R bloggers? click here if you have a blog, or here if you don’t.

In the new version (0.51.2) from nnetsauce (for Python, but also for R), available on PyPI and for conda, I removed jax and jaxlib (for GPU) from the standard version, because jaxlib is heavy.

It means that if you run GPUs with nnetsauce (as in https://www.researchgate.net/publication/382589729_Probabilistic_Forecasting_with_nnetsauce_using_Density_Estimation_Bayesian_inference_Conformal_prediction_and_Vine_copulas), you might want to install jax explicitly:

pip install nnetsauce[jax]

or

uv pip install nnetsauce[jax]

or

conda install -c conda-forge nnetsauce jax jaxlib

Nasty leave a comment for the author: follow the link and comment on their blog: The web page of T. Moudiki – R.

R-bloggers.com offers daily email updates about R news and tutorials on learning R and many other topics. To post or search for an R/data science job, click here.


Want to share your content on R bloggers? click here if you have a blog, or here if you don’t.


#nnetsauce #jax #GPU #acceleration #bloggers

Similar Posts

Leave a Reply

Your email address will not be published. Required fields are marked *