<p dir="auto">Please:</p>
<ul class="contains-task-list">
<li class="task-list-item"><input type="checkbox" id="" disabled="" class="task-list-item-checkbox" checked=""> Check for duplicate requests.</li>
<li class="task-list-item"><input type="checkbox" id="" disabled="" class="task-list-item-checkbox" checked=""> Describe your goal, and if possible provide a code snippet with a motivating example.</li>
</ul>
<p dir="auto">I think the documentation for the <code class="notranslate">jax.random.gamma</code> random variable (sampler) could possibly be clearer. In particular, letting the user know that the standard form is implemented and how to incorporate the rate (or alternatively scale) parameter in their samples. Maybe an addendum along the lines of</p>
<p dir="auto"><em>"we implement the standard gamma density. Dividing the sample by the rate is equivalent to sampling from</em> <math-renderer class="js-inline-math" style="display: inline" data-static-url="https://github.githubassets.com/static" data-run-id="7af9ac75c1357d6d22994502ac195ca5">$\text{Gamma}(\alpha, \text{rate})$</math-renderer><em>. And multiplying the sample by the scale is equivalent to sampling from</em> <math-renderer class="js-inline-math" style="display: inline" data-static-url="https://github.githubassets.com/static" data-run-id="7af9ac75c1357d6d22994502ac195ca5">$\text{Gamma}(\alpha, \text{scale})$</math-renderer>."</p>
<p dir="auto">The discussion for this started <a href="https://github.com/google/jax/pull/551" data-hovercard-type="pull_request" data-hovercard-url="/google/jax/pull/551/hovercard">here</a> and I'm mentioning <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/users/mattjj/hovercard" data-octo-click="hovercard-link-click" data-octo-dimensions="link_type:self" href="https://github.com/mattjj">@mattjj</a> for visibility. Thanks!</p>
<p dir="auto">Please:</p>
<ul class="contains-task-list">
<li class="task-list-item"><input type="checkbox" id="" disabled="" class="task-list-item-checkbox" checked=""> Check for duplicate issues.</li>
<li class="task-list-item"><input type="checkbox" id="" disabled="" class="task-list-item-checkbox" checked=""> Provide a complete example of how to reproduce the bug, wrapped in triple backticks like this:</li>
</ul>
<div class="highlight highlight-source-shell notranslate position-relative overflow-auto" dir="auto" data-snippet-clipboard-copy-content="# On Ubuntu 20.04.3, conda-forge channel only
conda create -n test python=3.9 openssl=1 jax jaxlib=0.3.0
conda activate test
python -c 'import jax'"><pre class="notranslate"><span class="pl-c"><span class="pl-c">#</span> On Ubuntu 20.04.3, conda-forge channel only</span>
conda create -n <span class="pl-c1">test</span> python=3.9 openssl=1 jax jaxlib=0.3.0
conda activate <span class="pl-c1">test</span>
python -c <span class="pl-s"><span class="pl-pds">'</span>import jax<span class="pl-pds">'</span></span></pre></div>
<ul class="contains-task-list">
<li class="task-list-item"><input type="checkbox" id="" disabled="" class="task-list-item-checkbox" checked=""> If applicable, include full error messages/tracebacks.</li>
</ul>
<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clipboard-copy-content="[libprotobuf ERROR google/protobuf/descriptor_database.cc:641] File already exists in database: tensorflow/compiler/xla/xla_data.proto
[libprotobuf FATAL google/protobuf/descriptor.cc:2021] CHECK failed: GeneratedDatabase()-&gt;Add(encoded_file_descriptor, size):
terminate called after throwing an instance of 'google::protobuf::FatalException'
  what():  CHECK failed: GeneratedDatabase()-&gt;Add(encoded_file_descriptor, size):
Aborted (core dumped)"><pre class="notranslate"><code class="notranslate">[libprotobuf ERROR google/protobuf/descriptor_database.cc:641] File already exists in database: tensorflow/compiler/xla/xla_data.proto
[libprotobuf FATAL google/protobuf/descriptor.cc:2021] CHECK failed: GeneratedDatabase()-&gt;Add(encoded_file_descriptor, size):
terminate called after throwing an instance of 'google::protobuf::FatalException'
  what():  CHECK failed: GeneratedDatabase()-&gt;Add(encoded_file_descriptor, size):
Aborted (core dumped)
</code></pre></div>