Update documentation and requirements; remove old examples and enhance installation instructions

Author: rogeriojorge

README.md

@@ -36,7 +36,7 @@
 - [Acknowledgments](#acknowledgments)
 
 ## Overview
-ESSOS is an open-source project in Python that uses JAX to optimize stellarator coils. Optimization can be applied to several objectives, such as alpha particle confinement, plasma boundaries and magnetic field equilibria. It leverages automatic differentiation and efficient numerical methods to streamline optimization efforts, creating a specialized and fast numerical tool for optimizing force-free stellarator equilibria. It can be imported in a Python script using the **essos** package, or run directly in the command line as `essos`. To install it, use
+ESSOS is an open-source project in Python that uses JAX to optimize stellarator coils. Optimization can be applied to several objectives, such as alpha particle confinement, plasma boundaries and magnetic field equilibria. It leverages automatic differentiation and efficient numerical methods to streamline optimization efforts, creating a specialized and fast numerical tool for optimizing force-free stellarator equilibria. It is parallelized using JAX's sharding tools. It can be imported in a Python script using the **essos** package, or run directly in the command line as `essos`. To install it, use
 
    ```sh
    pip install essos

docs/getting_started.rst

@@ -7,27 +7,21 @@ Installation
 ------------
 
 To use ESSOS, there is no need to install it.
-You can simply clone the repository and install the dependencies using the following command:
+You can simply clone the repository and install it using the following command:
 
 .. code-block:: console
 
     $ git clone https://github.com/uwplasma/ESSOS.git
     $ cd ESSOS
-    $ pip install -r requirements.txt
-
-these are
-
-    - jax
-    - jax_tqdm
-    - matplotlib
+    $ pip install .
 
 Run an example
 --------------
 
-To run the main example, use the following command:
+To run the one of the examples, use the following command:
 
 .. code-block:: console
 
-    python main.py
+    python examples/trace_fieldlines_coils.py
 
 More examples are in the `examples` folder.

docs/requirements.txt

@@ -1,5 +1,8 @@
 jax
-jax_tqdm
+jaxlib
+diffrax
+scipy
+netcdf4
 matplotlib
 sphinx
 sphinx-autoapi