Esc

Ask AIAnswers may be inaccurate; check the linked pages.Esc

Ask anything about these docs, like how to get started or what a function does.

Chronax Docs

Guides

Chronax
Nixtla `neuralforecast` baselines (per model)
Benchmarking Suite Documentation
Code Architecture & Execution Flow
Neural benchmark harness
DeepNPTS — JAX/Flax-NNX port into Chronax

API Reference

benchmarks.benchmark_suite
benchmarks.bitcn_benchmark
benchmarks.dilated_rnn_weight_parity
benchmarks.itransformer_benchmark
benchmarks.kan_weight_parity
benchmarks.neural.metrics
benchmarks.neural.registry
benchmarks.neural.run
benchmarks.neural.worker
benchmarks.patchtst_weight_parity
benchmarks.run_benchmark
benchmarks.softs_weight_parity
benchmarks.softssharp_weight_parity
benchmarks.vanillatransformer_benchmark
chronax.models
chronax.models.adida
chronax.models.arima
chronax.models.arima.arima
chronax.models.arima.auto_arima
chronax.models.autoformer
chronax.models.autoformer.data
chronax.models.autoformer.forecaster
chronax.models.autoformer.loss
chronax.models.autoformer.model
chronax.models.autoformer.train
chronax.models.base_forecaster
chronax.models.batched_forecaster
chronax.models.bitcn
chronax.models.bitcn.bitcn_losses
chronax.models.bitcn.bitcn_model
chronax.models.bitcn.bitcn_module
chronax.models.bitcn.bitcn_training
chronax.models.ces
chronax.models.croston_classic
chronax.models.deepar
chronax.models.deepar.data
chronax.models.deepar.forecaster
chronax.models.deepar.loss
chronax.models.deepar.model
chronax.models.deepar.train
chronax.models.deepnpts
chronax.models.deepnpts.deepnpts_losses
chronax.models.deepnpts.deepnpts_model
chronax.models.deepnpts.deepnpts_module
chronax.models.deepnpts.deepnpts_training
chronax.models.dilated_rnn
chronax.models.dilated_rnn.dilated_rnn_losses
chronax.models.dilated_rnn.dilated_rnn_model
chronax.models.dilated_rnn.dilated_rnn_module
chronax.models.dilated_rnn.dilated_rnn_scaler
chronax.models.dilated_rnn.dilated_rnn_training
chronax.models.dlinear
chronax.models.dlinear.dlinear_losses
chronax.models.dlinear.dlinear_model
chronax.models.dlinear.dlinear_module
chronax.models.dlinear.dlinear_scaler
chronax.models.dlinear.dlinear_training
chronax.models.ets
chronax.models.ets.auto_ets
chronax.models.ets.ets_backend
chronax.models.ets.ets_functions
chronax.models.ets.ets_model
chronax.models.fedformer
chronax.models.fedformer.data
chronax.models.fedformer.forecaster
chronax.models.fedformer.loss
chronax.models.fedformer.model
chronax.models.fedformer.train
chronax.models.garch
chronax.models.gru
chronax.models.gru.gru_losses
chronax.models.gru.gru_model
chronax.models.gru.gru_module
chronax.models.gru.gru_scaler
chronax.models.gru.gru_training
chronax.models.hint
chronax.models.hint.hint_model
chronax.models.historic_average
chronax.models.holt
chronax.models.holt_winters
chronax.models.imapa
chronax.models.informer
chronax.models.informer.informer_layers
chronax.models.informer.informer_losses
chronax.models.informer.informer_model
chronax.models.informer.informer_module
chronax.models.informer.informer_scaler
chronax.models.informer.informer_training
chronax.models.itransformer
chronax.models.itransformer.itransformer_losses
chronax.models.itransformer.itransformer_model
chronax.models.itransformer.itransformer_module
chronax.models.itransformer.itransformer_training
chronax.models.kan
chronax.models.kan.kan_losses
chronax.models.kan.kan_model
chronax.models.kan.kan_module
chronax.models.kan.kan_scaler
chronax.models.kan.kan_training
chronax.models.loess
chronax.models.lstm
chronax.models.lstm.lstm
chronax.models.lstm.test
chronax.models.mfles
chronax.models.mfles.auto_mfles
chronax.models.mfles.mfles
chronax.models.mlp
chronax.models.mlp.mlp_losses
chronax.models.mlp.mlp_model
chronax.models.mlp.mlp_module
chronax.models.mlp.mlp_scaler
chronax.models.mlp.mlp_training
chronax.models.mstl
chronax.models.naive
chronax.models.nbeats
chronax.models.nbeats.data
chronax.models.nbeats.forecaster
chronax.models.nbeats.loss
chronax.models.nbeats.model
chronax.models.nbeats.train
chronax.models.nlinear
chronax.models.nlinear.nlinear_losses
chronax.models.nlinear.nlinear_model
chronax.models.nlinear.nlinear_module
chronax.models.nlinear.nlinear_scaler
chronax.models.nlinear.nlinear_training
chronax.models.patchtst
chronax.models.patchtst.patchtst_losses
chronax.models.patchtst.patchtst_model
chronax.models.patchtst.patchtst_module
chronax.models.patchtst.patchtst_training
chronax.models.randomWalkWithDrift
chronax.models.rnn
chronax.models.rnn.data
chronax.models.rnn.forecaster
chronax.models.rnn.loss
chronax.models.rnn.model
chronax.models.rnn.train
chronax.models.seasonal_exponential_smoothing
chronax.models.seasonal_naive
chronax.models.seasonal_window_average
chronax.models.simple_exponential_smoothing
chronax.models.softs
chronax.models.softs.softs_losses
chronax.models.softs.softs_model
chronax.models.softs.softs_module
chronax.models.softs.softs_training
chronax.models.softssharp
chronax.models.softssharp.softssharp_losses
chronax.models.softssharp.softssharp_model
chronax.models.softssharp.softssharp_module
chronax.models.softssharp.softssharp_training
chronax.models.stemgnn
chronax.models.stemgnn.stemgnn_losses
chronax.models.stemgnn.stemgnn_model
chronax.models.stemgnn.stemgnn_module
chronax.models.stemgnn.stemgnn_scaler
chronax.models.stemgnn.stemgnn_training
chronax.models.stl
chronax.models.tbats
chronax.models.tbats.tbats_core
chronax.models.tbats.tbats_model
chronax.models.tcn
chronax.models.tcn.tcn_losses
chronax.models.tcn.tcn_model
chronax.models.tcn.tcn_module
chronax.models.tcn.tcn_scaler
chronax.models.tcn.tcn_training
chronax.models.tft
chronax.models.tft.tft_layers
chronax.models.tft.tft_losses
chronax.models.tft.tft_model
chronax.models.tft.tft_module
chronax.models.tft.tft_scaler
chronax.models.tft.tft_training
chronax.models.theta
chronax.models.theta.auto_theta
chronax.models.theta.theta_model
chronax.models.tide
chronax.models.tide.data
chronax.models.tide.forecaster
chronax.models.tide.loss
chronax.models.tide.model
chronax.models.tide.train
chronax.models.timemixer
chronax.models.timemixer.timemixer_losses
chronax.models.timemixer.timemixer_model
chronax.models.timemixer.timemixer_module
chronax.models.timemixer.timemixer_training
chronax.models.timexer
chronax.models.timexer.timexer_losses
chronax.models.timexer.timexer_model
chronax.models.timexer.timexer_module
chronax.models.timexer.timexer_training
chronax.models.tsb
chronax.models.tsmixer
chronax.models.tsmixer.data
chronax.models.tsmixer.forecaster
chronax.models.tsmixer.loss
chronax.models.tsmixer.model
chronax.models.tsmixer.train
chronax.models.vanillatransformer
chronax.models.vanillatransformer.vanillatransformer_losses
chronax.models.vanillatransformer.vanillatransformer_model
chronax.models.vanillatransformer.vanillatransformer_module
chronax.models.vanillatransformer.vanillatransformer_training
chronax.models.window_average
chronax.models.xlinear
chronax.models.xlinear.xlinear_losses
chronax.models.xlinear.xlinear_model
chronax.models.xlinear.xlinear_module
chronax.models.xlinear.xlinear_scaler
chronax.models.xlinear.xlinear_training
chronax.models.xlstm
chronax.models.xlstm.xlstm_backend
chronax.models.xlstm.xlstm_functions
chronax.models.xlstm.xlstm_model
chronax.utils
chronax.utils.conformal_intervals
chronax.utils.loss_functions
chronax.utils.plotting
chronax.utils.utils

`chronax.models.rnn`

Chronax RNN — JAX/Flax port of Nixtla NeuralForecast's RNN model.

On this page

chronax.models.rnn