I have a hierarchical regression model with two levels, countries and local areas. The country level has 84 different ids and there are 156.000 local areas. The total number of rows in the full dataset is about 2.500.000. When I fit the model to a subset of the data, all rows that has a local area i…

Max_steps with sampling_jax.sample_numpyro_nuts

ricardoV94 August 11, 2022, 4:05am 5

Should work for all parameters, but we only tested those explicitly.

Topic		Replies	Views
NUTS slows dramatically after exactly 200 samples Questions	2	738	August 25, 2019
How can I set the maximum tree depth for the NUTS method from the numpyro library? v5	3	457	June 9, 2023
What does the "maximum tree depth" warning mean? Questions	6	7203	December 4, 2019
Issues with NUTS performance for large scale problems Questions	1	766	January 30, 2019
NUTS sampler - trying to fix the step size (step_scale vs step_size?) Questions sampling	2	1194	January 26, 2022