Upload folder using huggingface_hub

README.md

@@ -0,0 +1,28 @@
+---
+library_name: mlx
+tags:
+- mlx
+- time-series
+- forecasting
+- tirex
+base_model: NX-AI/TiRex
+---
+
+# kunal732/TiRex-mlx
+
+This model was converted from [`NX-AI/TiRex`](https://huggingface.co/NX-AI/TiRex)
+using [MLX-Swift-TS](https://github.com/kunal732/MLX-Swift-TS).
+
+## Use with MLX-Swift-TS
+
+```swift
+import MLXTimeSeries
+
+let forecaster = try await TimeSeriesForecaster.loadFromHub(id: "kunal732/TiRex-mlx")
+let input = TimeSeriesInput.univariate(historicalValues)
+let prediction = forecaster.forecast(input: input, predictionLength: 64)
+```
+
+## Original Model
+
+[NX-AI/TiRex](https://huggingface.co/NX-AI/TiRex)

config.json

@@ -0,0 +1,26 @@
+{
+  "model_type": "tirex",
+  "ts_model_class": "TiRexModel",
+  "hidden_size": 512,
+  "num_layers": 12,
+  "input_format": "patches",
+  "output_format": "quantiles",
+  "context_length": 2048,
+  "prediction_length": 64,
+  "embedding_dim": 512,
+  "input_patch_size": 32,
+  "output_patch_size": 32,
+  "num_heads": 4,
+  "ffn_proj_factor": 2.6667,
+  "quantiles": [
+    0.1,
+    0.2,
+    0.3,
+    0.4,
+    0.5,
+    0.6,
+    0.7,
+    0.8,
+    0.9
+  ]
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:00c8ddee8aba9592a465428e85e7369c42ea262fe25fa3ba03cfd4726ddbd933
+size 70598937