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

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -20
README.md CHANGED
@@ -22,32 +22,32 @@ Running under the **JiRackTrain** pipeline on the enterprise infrastructure clus
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
 
 
 
 
 
22
  ## Model Details
23
 
24
  - **Model Name**: TernaryTransformer3B (3.3 Billion Parameters)
25
+ - **Architecture**: TernaryTransformer (custom ternary bit-response logic)
26
+ - **Tokenizer**: [JiRack Pro Tokenizer (128K)](https://huggingface.co/CMSManhattan/JiRack-Pro-Tokenizer-128K) — 347 active language editions of Wikipedia + specialized robotic action tokens
 
 
27
  - **Pre-training Dataset**: [JiRack-Pretrain-Dataset](https://huggingface.co/datasets/CMSManhattan/JiRack-Pretrain-Dataset)
 
28
 
29
+ ## Key Metrics (Step ~838/1000 on Shard 1)
30
 
31
+ - **Iteration Speed**: ~242–352 seconds per iteration (network tensor loading)
32
+ - **Current Loss**: 2.9678
33
+ - **Moving Average Loss**: 6.0920
34
+ - **Perplexity (PPL)**: 442.33
35
 
36
+ Engineers anticipate a **steep drop** in the perplexity curve within the first 250 iterations of Shard 1 as the ternary weights align with the tokenizer's token distributions.
37
 
38
+ ## Intended Use
39
 
40
+ This foundational model is being developed specifically for **robotics applications**:
 
 
 
41
 
42
+ - Real-time control policies
43
+ - Multimodal reasoning (text + vision + action tokens)
44
+ - Edge deployment with ternary efficiency
45
+ - Low-latency physical interaction loops
46
 
47
+ ## Monitoring & Updates
 
 
 
 
48
 
49
+ Training is actively progressing across all **7 shards**. Follow this repository or the linked tokenizer/dataset cards for checkpoint releases, evaluation results, and fine-tuned robotic variants.
50
+
51
+ ---
52
+
53
+ **Stay tuned** — the ternary robotics revolution is just getting started! 🚀