diffsky 0.1.1__tar.gz → 0.1.2__tar.gz

{diffsky-0.1.1 → diffsky-0.1.2}/.github/workflows/linting.yml

@@ -16,7 +16,7 @@ jobs:
 
       - uses: conda-incubator/setup-miniconda@v2
         with:
-          python-version: 3.9
+          python-version: 3.11
           channels: conda-forge,defaults
           channel-priority: strict
           show-channel-urls: true

diffsky-0.1.2/.github/workflows/monthly-warning-test.yml

@@ -0,0 +1,58 @@
+name: Test for Warnings
+
+on:
+  workflow_dispatch: null
+  schedule:
+    # Runs "First of every month at 3:15am Central"
+    - cron: '15 8 1 * *'
+
+jobs:
+  tests:
+    name: tests
+    runs-on: "ubuntu-latest"
+
+    steps:
+      - uses: actions/checkout@v2
+        with:
+          fetch-depth: 0
+
+      - uses: conda-incubator/setup-miniconda@v2
+        with:
+          python-version: 3.11
+          channels: conda-forge,defaults
+          channel-priority: strict
+          show-channel-urls: true
+          miniforge-version: latest
+          miniforge-variant: Mambaforge
+          use-mamba: true
+
+      - name: configure conda and install code
+      # Test against current main branch of all diff+ dependencies
+        shell: bash -l {0}
+        run: |
+          conda config --set always_yes yes
+          mamba install --quiet \
+            --file=requirements.txt
+          mamba install -y -q \
+            flake8 \
+            pytest \
+            pytest-xdist \
+            pytest-cov \
+            pip \
+            setuptools \
+            "setuptools_scm>=7,<8" \
+            matplotlib \
+            python-build
+          pip uninstall diffmah --yes
+          pip uninstall diffstar --yes
+          pip uninstall dsps --yes
+          pip install --no-deps git+https://github.com/ArgonneCPAC/diffmah.git
+          pip install --no-deps git+https://github.com/ArgonneCPAC/diffstar.git
+          pip install --no-deps git+https://github.com/ArgonneCPAC/dsps.git
+          python -m pip install --no-build-isolation --no-deps -e .
+
+      - name: test that no warnings are raised
+        shell: bash -l {0}
+        run: |
+          export PYTHONWARNINGS=error
+          pytest -v diffsky --cov --cov-report=xml

{diffsky-0.1.1 → diffsky-0.1.2}/.github/workflows/tests.yml

@@ -1,4 +1,4 @@
-name: tests
+name: test-latest-diffstuff-releases
 
 on:
   push:
@@ -18,7 +18,7 @@ jobs:
 
       - uses: conda-incubator/setup-miniconda@v2
         with:
-          python-version: 3.9
+          python-version: 3.11
           channels: conda-forge,defaults
           channel-priority: strict
           show-channel-urls: true
@@ -27,6 +27,7 @@ jobs:
           use-mamba: true
 
       - name: configure conda and install code
+      # Test against conda-forge for all dependencies except use main branch for dsps
         shell: bash -l {0}
         run: |
           conda config --set always_yes yes
@@ -40,7 +41,10 @@ jobs:
             pip \
             setuptools \
             "setuptools_scm>=7,<8" \
+            matplotlib \
             python-build
+          pip uninstall dsps --yes
+          pip install --no-deps git+https://github.com/ArgonneCPAC/dsps.git
           python -m pip install --no-build-isolation --no-deps -e .
 
       - name: test

diffsky-0.1.2/.github/workflows/tests_cron.yml

@@ -0,0 +1,88 @@
+name: test-main-branch-diffstuff
+
+on:
+  workflow_dispatch: null
+  schedule:
+    # Runs "every Monday & Thursday at 3:05am Central"
+    - cron: '5 8 * * 1,4'
+  push:
+    branches:
+      - main
+  pull_request: null
+
+jobs:
+  tests:
+    name: tests
+    runs-on: "ubuntu-latest"
+
+    steps:
+      - uses: actions/checkout@v2
+        with:
+          fetch-depth: 0
+
+      - uses: conda-incubator/setup-miniconda@v2
+        with:
+          python-version: 3.11
+          channels: conda-forge,defaults
+          channel-priority: strict
+          show-channel-urls: true
+          miniforge-version: latest
+          miniforge-variant: Mambaforge
+          use-mamba: true
+
+      - name: configure conda and install code
+      # Test against current main branch of all diff+ dependencies
+        shell: bash -l {0}
+        run: |
+          conda config --set always_yes yes
+          mamba install --quiet \
+            --file=requirements.txt
+          mamba install -y -q \
+            flake8 \
+            pytest \
+            pytest-xdist \
+            pytest-cov \
+            pip \
+            setuptools \
+            "setuptools_scm>=7,<8" \
+            matplotlib \
+            python-build
+          pip uninstall diffmah --yes
+          pip uninstall diffstar --yes
+          pip uninstall dsps --yes
+          pip install --no-deps git+https://github.com/ArgonneCPAC/diffmah.git
+          pip install --no-deps git+https://github.com/ArgonneCPAC/diffstar.git
+          pip install --no-deps git+https://github.com/ArgonneCPAC/dsps.git
+          python -m pip install --no-build-isolation --no-deps -e .
+
+      - name: test
+        shell: bash -l {0}
+        run: |
+          pytest -v diffsky --cov --cov-report=xml
+
+      - name: Upload coverage reports to Codecov
+        uses: codecov/codecov-action@v3
+
+      - name: test versions
+        shell: bash -el {0}
+        run: |
+          pip uninstall diffsky --yes
+          [[ $(python setup.py --version) != "0.0.0" ]] || exit 1
+
+          rm -rf dist/*
+          python setup.py sdist
+          pip install --no-deps --no-build-isolation dist/*.tar.gz
+          pushd ..
+          python -c "import diffsky; assert diffsky.__version__ != '0.0.0'"
+          popd
+          pip uninstall diffsky --yes
+
+          rm -rf dist/*
+          python -m build --sdist . --outdir dist
+          pip install --no-deps --no-build-isolation dist/*.tar.gz
+          pushd ..
+          python -c "import diffsky; assert diffsky.__version__ != '0.0.0'"
+          popd
+          pip uninstall diffsky --yes
+
+          python -m pip install -v --no-deps --no-build-isolation -e .

{diffsky-0.1.1 → diffsky-0.1.2}/CHANGES.rst

@@ -1,3 +1,8 @@
+0.1.2 (2024-10-25)
+-------------------
+- Update calls to diffmah v0.6.1 and diffstar v0.3.2
+
+
 0.1.1 (2023-10-04)
 -------------------
 - Update calls to dsps v0.3.4

{diffsky-0.1.1 → diffsky-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: diffsky
-Version: 0.1.1
+Version: 0.1.2
 Summary: Library for differentiable generation of synthetic skies
 Author-email: Andrew Hearin <ahearin@anl.gov>
 License: BSD 3-Clause License
@@ -35,15 +35,15 @@ License: BSD 3-Clause License
         
 Project-URL: home, https://github.com/ArgonneCPAC/diffsky
 Classifier: Programming Language :: Python :: 3
-Requires-Python: >=3.9
+Requires-Python: >=3.11
 Description-Content-Type: text/x-rst
 License-File: LICENSE.rst
 Requires-Dist: numpy
 Requires-Dist: jax
-Requires-Dist: h5py
-Requires-Dist: diffmah
-Requires-Dist: diffstar
+Requires-Dist: diffmah>=0.6.1
+Requires-Dist: diffstar>=0.3.2
 Requires-Dist: dsps
+Requires-Dist: h5py
 
 diffsky
 ============
@@ -69,7 +69,7 @@ Conda environment
 ~~~~~~~~~~~~~~~~~
 For a typical development environment in conda-forge::
 
-    $ conda create -c conda-forge -n diffsky_env python=3.9 numpy jax pytest ipython jupyter matplotlib scipy h5py diffmah diffstar dsps diffsky
+    $ conda create -c conda-forge -n diffsky_env python=3.11 numpy jax pytest ipython jupyter matplotlib scipy h5py diffmah diffstar dsps diffsky
 
 
 Documentation

{diffsky-0.1.1 → diffsky-0.1.2}/README.rst

@@ -22,7 +22,7 @@ Conda environment
 ~~~~~~~~~~~~~~~~~
 For a typical development environment in conda-forge::
 
-    $ conda create -c conda-forge -n diffsky_env python=3.9 numpy jax pytest ipython jupyter matplotlib scipy h5py diffmah diffstar dsps diffsky
+    $ conda create -c conda-forge -n diffsky_env python=3.11 numpy jax pytest ipython jupyter matplotlib scipy h5py diffmah diffstar dsps diffsky
 
 
 Documentation

{diffsky-0.1.1 → diffsky-0.1.2}/diffsky/__init__.py

@@ -1,5 +1,7 @@
 """
 """
+
 # flake8: noqa
 
 from ._version import __version__
+from .mass_functions.mc_diffmah_tpeak import mc_subhalos

diffsky-0.1.2/diffsky/_version.py

@@ -0,0 +1 @@
+__version__ = '0.1.2'

diffsky-0.1.2/diffsky/burstpop/__init__.py

@@ -0,0 +1,11 @@
+"""
+"""
+# flake8: noqa
+
+from .diffburstpop import (
+    DEFAULT_DIFFBURSTPOP_PARAMS,
+    DEFAULT_DIFFBURSTPOP_U_PARAMS,
+    calc_bursty_age_weights_from_diffburstpop_params,
+    get_bounded_diffburstpop_params,
+    get_unbounded_diffburstpop_params,
+)

diffsky-0.1.2/diffsky/burstpop/diagnostics/plot_fburstpop.py

@@ -0,0 +1,60 @@
+"""
+"""
+
+import numpy as np
+from matplotlib import pyplot as plt
+
+from ..fburstpop import DEFAULT_FBURSTPOP_PARAMS, get_lgfburst_from_fburstpop_params
+
+
+def make_fburstpop_comparison_plot(
+    params,
+    params2=DEFAULT_FBURSTPOP_PARAMS,
+    fname=None,
+    label1=r"${\rm new\ model}$",
+    label2=r"${\rm default\ model}$",
+):
+    """Make basic diagnostic plot of the model for Fburst
+
+    Parameters
+    ----------
+    params : namedtuple
+        Instance of fburstpop.FburstPopParams
+
+    params2 : namedtuple, optional
+        Instance of fburstpop.FburstPopParams
+        Default is set by DEFAULT_FBURSTPOP_PARAMS
+
+    fname : string, optional
+        filename of the output figure
+
+    """
+    nsm, nsfr = 250, 250
+    logsm_grid = np.linspace(7, 12, nsm)
+    logssfr_grid = np.linspace(-13, -8, nsfr)
+
+    X, Y = np.meshgrid(logsm_grid, logssfr_grid)
+
+    Z = get_lgfburst_from_fburstpop_params(params, X, Y)
+    Z2 = get_lgfburst_from_fburstpop_params(params2, X, Y)
+
+    fig, axes = plt.subplots(1, 2, figsize=(10, 4))
+    (ax0, ax1) = axes
+
+    pcm0 = ax0.pcolor(X, Y, Z, cmap="coolwarm_r", vmin=-4.5, vmax=-2.1)
+    fig.colorbar(pcm0, ax=ax0)
+
+    pcm1 = ax1.pcolor(X, Y, Z2, cmap="coolwarm_r", vmin=-4.5, vmax=-2.1)
+    fig.colorbar(pcm1, ax=ax1, label=r"${\rm lgFburst}$")
+    for ax in axes:
+        xlabel = ax.set_xlabel(r"$\log_{10}M_{\star}/M_{\odot}$")
+    ylabel = ax0.set_ylabel(r"${\rm \log_{10}sSFR}$")
+
+    ax0.set_title(label1)
+    ax1.set_title(label2)
+
+    if fname is not None:
+        fig.savefig(
+            fname, bbox_extra_artists=[xlabel, ylabel], bbox_inches="tight", dpi=200
+        )
+    return fig

diffsky-0.1.2/diffsky/burstpop/diagnostics/tests/test_plot_fburstpop.py

@@ -0,0 +1,15 @@
+"""
+"""
+
+import os
+
+from ..plot_fburstpop import DEFAULT_FBURSTPOP_PARAMS, make_fburstpop_comparison_plot
+
+_THIS_DRNAME = os.path.dirname(os.path.abspath(__file__))
+
+
+def test_make_freqburst_comparison_plot():
+    fn = os.path.join(_THIS_DRNAME, "dummy.png")
+    make_fburstpop_comparison_plot(DEFAULT_FBURSTPOP_PARAMS, fname=fn)
+    assert os.path.isfile(fn)
+    os.remove(fn)

diffsky-0.1.2/diffsky/burstpop/diffburstpop.py

@@ -0,0 +1,87 @@
+"""
+"""
+from collections import namedtuple
+
+from dsps.sfh.diffburst import BurstParams, calc_bursty_age_weights
+from jax import jit as jjit
+
+from .fburstpop import (
+    DEFAULT_FBURSTPOP_PARAMS,
+    get_bounded_fburstpop_params,
+    get_lgfburst_from_fburstpop_params,
+    get_unbounded_fburstpop_params,
+)
+from .tburstpop import (
+    DEFAULT_TBURSTPOP_PARAMS,
+    get_bounded_tburstpop_params,
+    get_tburst_params_from_tburstpop_params,
+    get_unbounded_tburstpop_params,
+)
+
+DiffburstPopParams = namedtuple(
+    "DiffburstPopParams", ["fburstpop_params", "tburstpop_params"]
+)
+DEFAULT_DIFFBURSTPOP_PARAMS = DiffburstPopParams(
+    DEFAULT_FBURSTPOP_PARAMS, DEFAULT_TBURSTPOP_PARAMS
+)
+_BURSTPOP_UPNAMES = [
+    key.replace("params", "u_params") for key in DEFAULT_DIFFBURSTPOP_PARAMS._fields
+]
+DiffburstPopUParams = namedtuple("DiffburstPopUParams", _BURSTPOP_UPNAMES)
+
+
+@jjit
+def get_bounded_diffburstpop_params(u_params):
+    u_fburstpop_params, u_tburstpop_params = u_params
+    bounded_tburstpop_params = get_bounded_tburstpop_params(u_tburstpop_params)
+    bounded_fburstpop_params = get_bounded_fburstpop_params(u_fburstpop_params)
+    diffburstpop_params = DiffburstPopParams(
+        bounded_fburstpop_params, bounded_tburstpop_params
+    )
+    return diffburstpop_params
+
+
+@jjit
+def get_unbounded_diffburstpop_params(params):
+    fburstpop_params, tburstpop_params = params
+    unbounded_fburstpop_params = get_unbounded_fburstpop_params(fburstpop_params)
+    unbounded_tburstpop_params = get_unbounded_tburstpop_params(tburstpop_params)
+    diffburstpop_u_params = DiffburstPopUParams(
+        unbounded_fburstpop_params, unbounded_tburstpop_params
+    )
+    return diffburstpop_u_params
+
+
+@jjit
+def calc_bursty_age_weights_from_diffburstpop_params(
+    diffburstpop_params, logsm, logssfr, ssp_lg_age_gyr, smooth_age_weights
+):
+    lgfburst = get_lgfburst_from_fburstpop_params(
+        diffburstpop_params.fburstpop_params, logsm, logssfr
+    )
+
+    tburst_params = get_tburst_params_from_tburstpop_params(
+        diffburstpop_params.tburstpop_params, logsm, logssfr
+    )
+    lgyr_peak, lgyr_max = tburst_params
+    burst_params = BurstParams(lgfburst, lgyr_peak, lgyr_max)
+
+    age_weights = calc_bursty_age_weights(
+        burst_params, smooth_age_weights, ssp_lg_age_gyr
+    )
+
+    return age_weights, burst_params
+
+
+def calc_bursty_age_weights_from_diffburstpop_u_params(
+    diffburstpop_u_params, logsm, logssfr, ssp_lg_age_gyr, smooth_age_weights
+):
+    diffburstpop_params = get_bounded_diffburstpop_params(diffburstpop_u_params)
+    args = diffburstpop_params, logsm, logssfr, ssp_lg_age_gyr, smooth_age_weights
+    age_weights, burst_params = calc_bursty_age_weights_from_diffburstpop_params(*args)
+    return age_weights, burst_params
+
+
+DEFAULT_DIFFBURSTPOP_U_PARAMS = DiffburstPopUParams(
+    *get_unbounded_diffburstpop_params(DEFAULT_DIFFBURSTPOP_PARAMS)
+)

diffsky-0.1.2/diffsky/burstpop/diffqburstpop.py

@@ -0,0 +1,97 @@
+"""
+"""
+
+from collections import namedtuple
+
+from dsps.sfh.diffburst import BurstParams, calc_bursty_age_weights
+from jax import jit as jjit
+
+from .fburstpop import (
+    DEFAULT_FBURSTPOP_PARAMS,
+    get_bounded_fburstpop_params,
+    get_lgfburst_from_fburstpop_params,
+    get_unbounded_fburstpop_params,
+)
+from .freqburst import (
+    DEFAULT_FREQBURST_PARAMS,
+    get_bounded_freqburst_params,
+    get_unbounded_freqburst_params,
+)
+from .tburstpop import (
+    DEFAULT_TBURSTPOP_PARAMS,
+    get_bounded_tburstpop_params,
+    get_tburst_params_from_tburstpop_params,
+    get_unbounded_tburstpop_params,
+)
+
+DiffburstPopParams = namedtuple(
+    "DiffburstPopParams", ["freqburst_params", "fburstpop_params", "tburstpop_params"]
+)
+DEFAULT_DIFFBURSTPOP_PARAMS = DiffburstPopParams(
+    DEFAULT_FREQBURST_PARAMS, DEFAULT_FBURSTPOP_PARAMS, DEFAULT_TBURSTPOP_PARAMS
+)
+_BURSTPOP_UPNAMES = [
+    key.replace("params", "u_params") for key in DEFAULT_DIFFBURSTPOP_PARAMS._fields
+]
+DiffburstPopUParams = namedtuple("DiffburstPopUParams", _BURSTPOP_UPNAMES)
+
+
+@jjit
+def get_bounded_diffburstpop_params(u_params):
+    u_freqburst_params, u_fburstpop_params, u_tburstpop_params = u_params
+    bounded_freqburst_params = get_bounded_freqburst_params(u_freqburst_params)
+    bounded_tburstpop_params = get_bounded_tburstpop_params(u_tburstpop_params)
+    bounded_fburstpop_params = get_bounded_fburstpop_params(u_fburstpop_params)
+    diffburstpop_params = DiffburstPopParams(
+        bounded_freqburst_params, bounded_fburstpop_params, bounded_tburstpop_params
+    )
+    return diffburstpop_params
+
+
+@jjit
+def get_unbounded_diffburstpop_params(params):
+    freqburst_params, fburstpop_params, tburstpop_params = params
+    unbounded_freqburst_params = get_unbounded_freqburst_params(freqburst_params)
+    unbounded_fburstpop_params = get_unbounded_fburstpop_params(fburstpop_params)
+    unbounded_tburstpop_params = get_unbounded_tburstpop_params(tburstpop_params)
+    diffburstpop_u_params = DiffburstPopUParams(
+        unbounded_freqburst_params,
+        unbounded_fburstpop_params,
+        unbounded_tburstpop_params,
+    )
+    return diffburstpop_u_params
+
+
+@jjit
+def calc_bursty_age_weights_from_diffburstpop_params(
+    diffburstpop_params, logsm, logssfr, ssp_lg_age_gyr, smooth_age_weights
+):
+    lgfburst = get_lgfburst_from_fburstpop_params(
+        diffburstpop_params.fburstpop_params, logsm, logssfr
+    )
+
+    tburst_params = get_tburst_params_from_tburstpop_params(
+        diffburstpop_params.tburstpop_params, logsm, logssfr
+    )
+    lgyr_peak, lgyr_max = tburst_params
+    burst_params = BurstParams(lgfburst, lgyr_peak, lgyr_max)
+
+    age_weights = calc_bursty_age_weights(
+        burst_params, smooth_age_weights, ssp_lg_age_gyr
+    )
+
+    return age_weights, burst_params
+
+
+def calc_bursty_age_weights_from_diffburstpop_u_params(
+    diffburstpop_u_params, logsm, logssfr, ssp_lg_age_gyr, smooth_age_weights
+):
+    diffburstpop_params = get_bounded_diffburstpop_params(diffburstpop_u_params)
+    args = diffburstpop_params, logsm, logssfr, ssp_lg_age_gyr, smooth_age_weights
+    age_weights, burst_params = calc_bursty_age_weights_from_diffburstpop_params(*args)
+    return age_weights, burst_params
+
+
+DEFAULT_DIFFBURSTPOP_U_PARAMS = DiffburstPopUParams(
+    *get_unbounded_diffburstpop_params(DEFAULT_DIFFBURSTPOP_PARAMS)
+)

diffsky-0.1.2/diffsky/burstpop/fburstpop.py

@@ -0,0 +1,139 @@
+"""
+"""
+
+from collections import OrderedDict, namedtuple
+
+from jax import jit as jjit
+from jax import numpy as jnp
+from jax import vmap
+
+from ..utils import _inverse_sigmoid, _sigmoid
+
+LGSM_K = 5.0
+LGSSFR_K = 5.0
+
+DEFAULT_FBURSTPOP_PDICT = OrderedDict(
+    lgfburst_logsm_x0_x0=10.0,
+    lgfburst_logsm_x0_q=10.5,
+    lgfburst_logsm_x0_ms=9.5,
+    lgfburst_logsm_ylo_x0=-10.25,
+    lgfburst_logsm_ylo_q=-4.5,
+    lgfburst_logsm_ylo_ms=-4.5,
+    lgfburst_logsm_yhi_x0=-11.25,
+    lgfburst_logsm_yhi_q=-4.5,
+    lgfburst_logsm_yhi_ms=-4.5,
+)
+
+_LGSM_X0_BOUNDS = (8.0, 12.0)
+_LGSSFR_X0_BOUNDS = (-13.0, -7.0)
+_LGFBURST_BOUNDS = (-6.0, -2.0)
+FBURSTPOP_BOUNDS_PDICT = OrderedDict(
+    lgfburst_logsm_x0_x0=_LGSM_X0_BOUNDS,
+    lgfburst_logsm_x0_q=_LGSM_X0_BOUNDS,
+    lgfburst_logsm_x0_ms=_LGSM_X0_BOUNDS,
+    lgfburst_logsm_ylo_x0=_LGSSFR_X0_BOUNDS,
+    lgfburst_logsm_ylo_q=_LGFBURST_BOUNDS,
+    lgfburst_logsm_ylo_ms=_LGFBURST_BOUNDS,
+    lgfburst_logsm_yhi_x0=_LGSSFR_X0_BOUNDS,
+    lgfburst_logsm_yhi_q=_LGFBURST_BOUNDS,
+    lgfburst_logsm_yhi_ms=_LGFBURST_BOUNDS,
+)
+
+FburstPopParams = namedtuple("FburstPopParams", DEFAULT_FBURSTPOP_PDICT.keys())
+_FBURSTPOP_UPNAMES = ["u_" + key for key in DEFAULT_FBURSTPOP_PDICT.keys()]
+FburstPopUParams = namedtuple("FburstPopUParams", _FBURSTPOP_UPNAMES)
+
+
+DEFAULT_FBURSTPOP_PARAMS = FburstPopParams(**DEFAULT_FBURSTPOP_PDICT)
+FBURSTPOP_PBOUNDS = FburstPopParams(**FBURSTPOP_BOUNDS_PDICT)
+
+
+@jjit
+def get_lgfburst_from_fburstpop_u_params(fburstpop_u_params, logsm, logssfr):
+    fburstpop_params = get_bounded_fburstpop_params(fburstpop_u_params)
+    lgfburst = get_lgfburst_from_fburstpop_params(fburstpop_params, logsm, logssfr)
+    return lgfburst
+
+
+@jjit
+def get_lgfburst_from_fburstpop_params(fburstpop_params, logsm, logssfr):
+    lgfburst_logssfr_x0 = _sigmoid(
+        logsm,
+        fburstpop_params.lgfburst_logsm_x0_x0,
+        LGSM_K,
+        fburstpop_params.lgfburst_logsm_ylo_x0,
+        fburstpop_params.lgfburst_logsm_yhi_x0,
+    )
+    lgfburst_logssfr_q = _sigmoid(
+        logsm,
+        fburstpop_params.lgfburst_logsm_x0_q,
+        LGSM_K,
+        fburstpop_params.lgfburst_logsm_ylo_q,
+        fburstpop_params.lgfburst_logsm_yhi_q,
+    )
+    lgfburst_logssfr_ms = _sigmoid(
+        logsm,
+        fburstpop_params.lgfburst_logsm_x0_ms,
+        LGSSFR_K,
+        fburstpop_params.lgfburst_logsm_ylo_ms,
+        fburstpop_params.lgfburst_logsm_yhi_ms,
+    )
+
+    lgfburst = _sigmoid(
+        logssfr,
+        lgfburst_logssfr_x0,
+        LGSSFR_K,
+        lgfburst_logssfr_q,
+        lgfburst_logssfr_ms,
+    )
+    return lgfburst
+
+
+@jjit
+def _get_bounded_fburstpop_param(u_param, bound):
+    lo, hi = bound
+    mid = 0.5 * (lo + hi)
+    return _sigmoid(u_param, mid, 0.1, lo, hi)
+
+
+@jjit
+def _get_unbounded_fburstpop_param(param, bound):
+    lo, hi = bound
+    mid = 0.5 * (lo + hi)
+    return _inverse_sigmoid(param, mid, 0.1, lo, hi)
+
+
+_C = (0, 0)
+_get_bounded_fburstpop_params_kern = jjit(
+    vmap(_get_bounded_fburstpop_param, in_axes=_C)
+)
+_get_unbounded_fburstpop_params_kern = jjit(
+    vmap(_get_unbounded_fburstpop_param, in_axes=_C)
+)
+
+
+@jjit
+def get_bounded_fburstpop_params(u_params):
+    u_params = jnp.array([getattr(u_params, u_pname) for u_pname in _FBURSTPOP_UPNAMES])
+    params = _get_bounded_fburstpop_params_kern(
+        jnp.array(u_params), jnp.array(FBURSTPOP_PBOUNDS)
+    )
+    fburstpop_params = FburstPopParams(*params)
+    return fburstpop_params
+
+
+@jjit
+def get_unbounded_fburstpop_params(params):
+    params = jnp.array(
+        [getattr(params, pname) for pname in DEFAULT_FBURSTPOP_PARAMS._fields]
+    )
+    u_params = _get_unbounded_fburstpop_params_kern(
+        jnp.array(params), jnp.array(FBURSTPOP_PBOUNDS)
+    )
+    fburstpop_u_params = FburstPopUParams(*u_params)
+    return fburstpop_u_params
+
+
+DEFAULT_FBURSTPOP_U_PARAMS = FburstPopUParams(
+    *get_unbounded_fburstpop_params(DEFAULT_FBURSTPOP_PARAMS)
+)