Robotics
multilingual
ternary
multimodal
pretraining
jirack
ternarytransformer
kgrabko commited on
Commit
e30e939
·
verified ·
1 Parent(s): 1ab1a06

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +53 -0
README.md ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: multilingual
3
+ tags:
4
+ - ternary
5
+ - robotics
6
+ - multimodal
7
+ - pretraining
8
+ - jirack
9
+ - ternarytransformer
10
+ license: apache-2.0
11
+ datasets:
12
+ - CMSManhattan/JiRack-Pretrain-Dataset
13
+ inference: false
14
+ ---
15
+
16
+ # JiRack Robotics - TernaryTransformer3B (Pre-training Phase)
17
+
18
+ **JiRack Robotics** has officially kicked off the **multi-shard pre-training phase** for its latest **3.3B parameter robotics model**.
19
+
20
+ Running under the **JiRackTrain** pipeline on the enterprise infrastructure cluster (`root@jirack1`), the training initializes the next-generation **TernaryTransformer3B** architecture, tightly coupled with the advanced **JiRack Pro Tokenizer**.
21
+
22
+ ## Model Details
23
+
24
+ - **Model Name**: TernaryTransformer3B (3.3 Billion Parameters)
25
+ - **Architecture**: TernaryTransformer (custom ternary bit-response logic with GQA + MoE scaling foundations)
26
+ - **Tokenizer**: [JiRack Pro Tokenizer (128K)](https://huggingface.co/CMSManhattan/JiRack-Pro-Tokenizer-128K)
27
+ - 347 active language editions of Wikipedia
28
+ - Specialized robotic action tokens + traditional text/vision embeddings
29
+ - **Pre-training Dataset**: [JiRack-Pretrain-Dataset](https://huggingface.co/datasets/CMSManhattan/JiRack-Pretrain-Dataset)
30
+ - **Training Objective**: From-scratch pre-training using the exact same multimodal corpus used to optimize the JiRack Pro Tokenizer (perfect vocabulary alignment)
31
+
32
+ ## Why This Alignment Matters
33
+
34
+ By reusing the tokenizer's own training corpus for model pre-training, JiRack Robotics eliminates standard vocabulary bias found in most open-source LLMs. The ternary neural paths learn directly from a data distribution that maps perfectly to the token vocabulary boundaries. This is expected to deliver significantly higher deterministic accuracy for real-time robotic control policies, edge processing, and low-latency physical maneuver loops.
35
+
36
+ ## Training Setup & Technical Specifications
37
+
38
+ The training leverages the custom `train_jirack_accelerate.py` framework with the following key paradigms:
39
+
40
+ - **Memory Optimization**: Active **Gradient Checkpointing** across all TernaryTransformer3B layers (drastically reduced VRAM footprint)
41
+ - **Data Layout**: Sequential pipeline distributed across **7 cloud shards** (`jirack_pretrain_chunk_0.pt` through `jirack_pretrain_chunk_6.pt`)
42
+ - **Infrastructure**: Enterprise cluster with network-mounted storage (`/mnt/nfs_clientshare/JiRackTrain`)
43
+ - **Framework**: PyTorch + Accelerate (Python 3.12)
44
+
45
+ **Current Training Log Snapshot** (Shard 1/7):
46
+
47
+ ```bash
48
+ Shards found for processing: 7
49
+ Initializing JiRack architecture (3.3B) from your class...
50
+ -> Gradient Checkpointing successfully activated on TernaryTransformer3B.
51
+ Starting sequential training loop across shards...
52
+
53
+ [Shard 1/7] Loading over network: jirack_pretrain_chunk_0.pt