From 88ebe8961666f2aac8bd43731c5065d846654814 Mon Sep 17 00:00:00 2001 From: erfanzar Date: Tue, 28 May 2024 17:14:13 +0330 Subject: [PATCH] improving `documentations` --- docs/contributing.rst | 4 +- docs/{easydelstate.rst => easydelstate.md} | 2 +- docs/index.rst | 2 +- docs/{install.rst => install.md} | 53 +++++++++++++--------- 4 files changed, 35 insertions(+), 26 deletions(-) rename docs/{easydelstate.rst => easydelstate.md} (96%) rename docs/{install.rst => install.md} (53%) diff --git a/docs/contributing.rst b/docs/contributing.rst index 14b9ba38..e634022f 100644 --- a/docs/contributing.rst +++ b/docs/contributing.rst @@ -15,7 +15,7 @@ To contribute to EasyDeL, follow these steps: Code of Conduct ------ -Please adhere to the Apache Code of Conduct_ in all interactions related to EasyDeL. +Please adhere to the `Apache Code of Conduct `_ in all interactions related to EasyDeL. Reporting Bugs ------ @@ -41,5 +41,3 @@ License By contributing to EasyDeL, you agree that your contributions will be licensed under the Apache License, Version 2.0. Thank you for your interest in contributing to EasyDeL! We appreciate your support. - -.. _Conduct https://www.apache.org/foundation/policies/conduct.html \ No newline at end of file diff --git a/docs/easydelstate.rst b/docs/easydelstate.md similarity index 96% rename from docs/easydelstate.rst rename to docs/easydelstate.md index 105f1643..3169ef87 100644 --- a/docs/easydelstate.rst +++ b/docs/easydelstate.md @@ -1,4 +1,4 @@ -**EasyDeLState: A Snapshot of Your EasyDeL Model** +# **EasyDeLState: A Snapshot of Your EasyDeL Model** The `EasyDeLState` class acts like a comprehensive container that holds all the essential information about your EasyDeL model at a given point in time. Think of it as a snapshot of your model. It includes: diff --git a/docs/index.rst b/docs/index.rst index 980cb536..4d45bd75 100644 --- a/docs/index.rst +++ b/docs/index.rst @@ -99,7 +99,7 @@ Zare Chavoshi, Erfan. "EasyDeL, an open-source library, is specifically designed install.md contributing - easydelstate + easydelstate.md .. toctree:: diff --git a/docs/install.rst b/docs/install.md similarity index 53% rename from docs/install.rst rename to docs/install.md index 7a7944ea..32a72314 100644 --- a/docs/install.rst +++ b/docs/install.md @@ -1,29 +1,39 @@ -Installing EasyDeL -====== -EasyDeL uses FJFormer and JAX as main dependencies in order to run the scripts but there are some things that needs to be installed such as GO-lang to JAX specific platform installations, but you can simply install EasyDeL via pip: +# Installing EasyDeL -``` +EasyDeL uses FJFormer and JAX as main dependencies in order to run the scripts but there are some things that needs to +be installed such as GO-lang to JAX specific platform installations, but you can simply install EasyDeL via pip: + +```shell pip install easydel ``` -Installing Jax ------- -JAX uses XLA to compile and run your NumPy programs on GPUs and TPUs. Compilation happens under the hood by default, with library calls getting just-in-time compiled and executed. But JAX also lets you just-in-time compile your own Python functions into XLA-optimized kernels using a one-function API, jit. +or install from head +```shell +pip install git+https://github.com/erfanzar/EasyDeL.git -U -q +``` + +## Installing Jax +JAX uses XLA to compile and run your NumPy programs on GPUs and TPUs. Compilation happens under the hood by default, +with library calls getting just-in-time compiled and executed. But JAX also lets you just-in-time compile your own +Python functions into XLA-optimized kernels using a one-function API, jit. you can install other version too but easydel required at least version of 0.4.16 -TPU ------- +### TPU + +inorder to install jax on TPU Devices use following command + ```shell !pip install jax[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html -q ``` -GPU ------- +### GPU + +### CUDA-12 + +inorder to install jax on cuda 12 use following command -CUDA-12 ------- ```shell pip install --upgrade pip # CUDA 12 installation @@ -31,8 +41,10 @@ pip install --upgrade pip pip install --upgrade "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html ``` -CUDA-11 ------- +### CUDA-11 + +inorder to install jax on cuda 11 use following command + ```shell pip install --upgrade pip # CUDA 11 installation @@ -40,19 +52,18 @@ pip install --upgrade pip pip install --upgrade "jax[cuda11_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html ``` +#### Installing GO -Installing GO ------- Note this Library needs golang to run (for some tracking stuff on TPU/GPU/CPU) -Ubuntu GO installation ------- + +#### Ubuntu GO installation ```shell sudo apt-get update && apt-get upgrade -y sudo apt-get install golang -y ``` -Manjaro/Arch GO installation ------- + +#### Manjaro/Arch GO installation ```shell sudo pacman -Syyuu go