As a user of AI Notebooks/data scientist,
I would like to have access to Jax as a framework
in order to work on numerical functions transformation with the ability to have access to new features such as auto-vectorization, auto-differentiation, auto-parallelization and just-in-time compilation.
Acceptance criteria
Ability to select Jax as a framework in AI Notebooks with:
Ubuntu 22.04
latest stable Conda version
latest stable NVIDIA CUDA Compiler Driver DEVEL version and appropriate solver
latest stable Miniconda version
latest stable Jax version
Python v3.11
Jax framework can be used with Jupyterlab/Jupyterlab Colab/VS Code latest versions
Compliant with CPU/GPU
Ability to launch through UI, CLI and API
Follow, vote and give your feedback
You can follow this task with the notification on the right tab. Ask us anything here in the comments below, and vote with emojis for most requested items! 👍 to vote for this issue
User story
As a user of AI Notebooks/data scientist, I would like to have access to Jax as a framework
in order to work on numerical functions transformation with the ability to have access to new features such as auto-vectorization, auto-differentiation, auto-parallelization and just-in-time compilation.
Acceptance criteria
Follow, vote and give your feedback
You can follow this task with the notification on the right tab. Ask us anything here in the comments below, and vote with emojis for most requested items! 👍 to vote for this issue
Discuss on Discord
Feel free to discuss with us on https://discord.gg/ovhcloud