Metadata-Version: 2.4
Name: neural-chemistry
Version: 1.0.6
Summary: Add your description here
Author-email: Andres Asensio Ramos <aasensio@iac.es>
License-File: LICENSE
Requires-Python: >=3.10
Requires-Dist: numpy
Requires-Dist: requests>=2.32.4
Requires-Dist: scipy
Requires-Dist: torch>=2.2
Requires-Dist: tqdm
Description-Content-Type: text/markdown

# Neural emulator for interstellar chemistry

This repository contains the code for using a conditional neural field
as an emulator for instellar chemistry.

## Installation

It should be enough to do:

    pip install neural_chemistry

and all requirements should be installed.

### Dependencies
In case it fails, note that this repository makes use of PyTorch and other Python packages. You can have them
installed in your system, but we recommend to follow these instructions to
have a working environment with everything needed for running the code.

Install Miniconda in your system. Go to https://docs.conda.io/projects/miniconda/en/latest/ and download the executable file for your system. For instance, for a typical Linux system, you should do:

    wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh

or 

    curl -lO https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh

Install Miniconda

    chmod +x Miniconda3-latest-Linux-x86_64.sh
    ./Miniconda3-latest-Linux-x86_64.sh

and follow the instructions for selecting the directory.

Create an environment and install the packages:

    conda create -n chemistry python=3.10
    conda activate chemistry
    conda install -c conda-forge numpy tqdm 

Now install PyTorch. Go to https://pytorch.org/ and select from the properties of 
your system from the matrix (Linux/Mac/Windows, CPU/GPU, conda/pip). If you have an 
NVIDIA GPU card in your system, you can take advantage of the acceleration that it
provides (around a factor 10 faster). For instance, in a system with an NVIDIA GPU
card with CUDA drivers on version 12.1, you can install PyTorch using:

    conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

or

    pip3 install torch torchvision torchaudio

if using `pip` for the installation of packages.

In a CPU-only system:

    conda install pytorch torchvision torchaudio cpuonly -c pytorch

or

    pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

if using `pip` for the installation of packages.

To make then a local installation, simply clone the repository and type

    pip install -e .

## New version

An example of how to run the code is shown in `example2.py`, that we reproduce here:

    # Define the number of random models 
    
    n_models = 32
    batch_size = 32

    mols = [0, 1, 2, 3, 4, 5]

    nh = 10.0**np.random.uniform(np.log10(1e4), np.log10(1e7), size=n_models)
    T = np.random.uniform(10.0, 80.0, size=n_models)
    crir = 10.0**np.random.uniform(np.log10(1e-17), np.log10(1e-15), size=n_models)
    sulfur = 10.0**np.random.uniform(np.log10(7.5e-8), np.log10(1.5e-5), size=n_models)
    uv_flux = 10.0**np.random.uniform(np.log10(0.1), np.log10(1e4), size=n_models)
    Av = np.random.uniform(0.0, 18.0, size=n_models)
    
    t = np.logspace(0, 7, 120)

    net = neural_chemistry.ChemistryEmulator2(gpu=0, verbose=True)
    abundance = net.evaluate(t, T, nh, crir, sulfur, uv_flux, Av, batch_size=batch_size, species=None)

    abundance_subset = net.evaluate(t, T, nh, crir, sulfur, uv_flux, Av, batch_size=batch_size, species=[0,1,2,3,4,5])


We choose 64 models with random properties inside the range of validity. Select the output times and call the emulator. The `batch_size` defines the number of models that will be computed in parallel. This depends on the amount of memory you have but it can be a large number. The keyword `species` indicates a list of the indices of which species you
want to compute, from the list of 250 species that can be found in `list_molecules:20ene24.txt`. If abstent or
set to `None`, all species are computed.

## Old version

An example of how to run the old code is shown in `example.py`, that we reproduce here:

    # Define the number of random models 
    n_models = 64
    batch_size = 32

    nh = 10.0**np.random.uniform(np.log10(1e4), np.log10(1e7), size=n_models)
    T = np.random.uniform(10.0, 80.0, size=n_models)
    crir = 10.0**np.random.uniform(np.log10(1e-17), np.log10(1e-15), size=n_models)
    sulfur = 10.0**np.random.uniform(np.log10(1.5e-5), np.log10(7.5e-8), size=n_models)
    uv_flux = 10.0**np.random.uniform(np.log10(0.1), np.log10(1e4), size=n_models)
    
    t = np.logspace(0, 7, 64)

    net = emulator.ChemistryEmulator(gpu=0, batch_size=32)
    abundance = net.evaluate(t, T, nh, crir, sulfur, uv_flux, batch_size=32, species=None)

## Weights

The weights are downloadad automatically the first time any of the emulators is used.