Chronax Documentation

A JAX-accelerated time-series forecasting library with a unified fit / predict interface.

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