xyaoNV commited on
Commit
602628b
·
verified ·
1 Parent(s): b9de2b4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +32 -23
README.md CHANGED
@@ -14,7 +14,7 @@ tags:
14
 
15
  ## Description:
16
 
17
- Arena-Dexsuite-Lift-RL-Newton-Task is a model trained for the object lifting task in one of IsaacLab-Arena Newton example workflows. The Kuka Allergo robot learns to pick up a primitive shape on the table and lift it to target position through reinforcement learning. The task uses a command-based goal specification, where the RL agent learns to reach sampled target poses.<br><br>
18
  This model is ready for non-commercial use only.
19
 
20
  ## Governing Terms
@@ -25,8 +25,8 @@ Global
25
 
26
  ### Release Date:
27
 
28
- GitHub: 03/30/2026 via [isaac-sim/IsaacLab-Arena](https://github.com/isaac-sim/IsaacLab-Arena/)
29
- Hugging Face: 03/30/2026 via [nvidia/Arena-Dexsuite-Lift-RL-Newton-Task](https://huggingface.co/nvidia/Arena-Dexsuite-Lift-RL-Newton-Task)
30
 
31
  ### Use Case:
32
  Researchers, Academics, Open-Source Community: AI-driven robotics research and algorithm development.
@@ -34,14 +34,14 @@ Developers: Integrate and customize AI for various robotic applications.
34
  Startups & Companies: Accelerate robotics development and reduce training costs.
35
 
36
  ## Reference(s):
37
- RSL‑RL: A Learning Library for Robotics Research, arXiv preprint (2025), arXiv:2509.10771
38
 
39
  ## Model Architecture:
40
 
41
  **Architecture Type:** Multi-layer Perceptron
42
 
43
  **Network Architecture:**
44
- The model is trained to move the Kuka end effector lifting the object to commanded target position. The model itself is a MLP network trained with on-policy PPO using RSL-RL PPO framework using Newton physics engine.
45
 
46
  **Number of Model Parameters:**
47
  105,871
@@ -56,43 +56,54 @@ The model is trained to move the Kuka end effector lifting the object to command
56
  **Input Parameters:**
57
  * Robot State: One-Dimensional (1D) - Floating number vector<br>
58
 
 
 
59
  ## Output:
60
  **Output Type(s):** End-effector target pose<br>
61
  **Output Format:** Continuous-value vectors<br>
62
  **Output Parameters:** [One-Dimensional (1D)] <br>
 
 
 
63
 
64
- Our AI models are designed and/or optimized to run on NVIDIA GPU-accelerated systems. By leveraging NVIDIA’s hardware (e.g. GPU cores) and software frameworks (e.g., CUDA libraries), the model achieves faster training and inference times compared to CPU-only solutions. <br>
65
 
66
  ## Model Version(s):
67
  Version 0.1.
68
 
69
  # Training, Testing, and Evaluation:
70
- **Training & Testing:** The model was trained completely in simulation (IsaacLab-Arena) using a on-policy Reinforcement Learning method.
71
 
72
- **Data Collection Method:** N/A (Reinforcement Learning Only)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
73
 
74
- **Labeling Method:** N/A
75
 
76
- **Properties:** There was no data collected or used for training. The model was trained completely in simulation (IsaacLab-Arena) using a on-policy Reinforcement Learning method. In this case, the data will be collected for a couple of episodes, then the algorithm will run gradient descent on these collected data to update the model weights. Once this step is done, the collected data will be discarded and a new batch of data will be collected and so on.
77
 
78
- As for the simulation environment, it contains the robot itself, a primitive sitting on a table, and a ground plane.
79
 
80
- All the training, testing, and evaluation was done in the same simulation environment. No data has been collected in any of these procedures.
81
 
 
82
 
83
  ## Inference:
84
  **Acceleration Engine(s):** PyTorch
85
 
86
  **Test Hardwares**
87
- All of the below:
88
- * NVIDIA Ampere
89
- * NVIDIA Blackwell
90
- * NVIDIA Jetson
91
- * NVIDIA Hopper
92
- * NVIDIA Lovelace
93
-
94
- **Supported Operating System(s):**
95
- * Linux
96
 
97
  ## Software Integration
98
 
@@ -108,8 +119,6 @@ All of the below:
108
 
109
  **Preferred/Supported Operating System(s):** Linux
110
 
111
- The integration of foundation and fine-tuned models into AI systems requires additional testing using use-case-specific data to ensure safe and effective deployment. Following the V-model methodology, iterative testing and validation at both unit and system levels are essential to mitigate risks, meet technical and functional requirements, and ensure compliance with safety and ethical standards before deployment.
112
-
113
  ## Ethical Considerations:
114
  NVIDIA believes Trustworthy AI is a shared responsibility and we have established policies and practices to enable development for a wide array of AI applications. When downloaded or used in accordance with our terms of service, developers should work with their internal model team to ensure this model meets requirements for the relevant industry and use case and addresses unforeseen product misuse.
115
 
 
14
 
15
  ## Description:
16
 
17
+ Arena-Dexsuite-Lift-RL-Newton-Task is a model trained for the object lifting task in one of IsaacLab-Arena Newton example workflows. The Kuka Allegro robot learns to pick up primitive shapes (e.g., cube-shaped objects) on a table and lift them to target positions through reinforcement learning. The task uses a command-based goal specification, where the RL agent learns to reach sampled target poses.<br><br>
18
  This model is ready for non-commercial use only.
19
 
20
  ## Governing Terms
 
25
 
26
  ### Release Date:
27
 
28
+ GitHub: 04/09/2026 via [isaac-sim/IsaacLab-Arena](https://github.com/isaac-sim/IsaacLab-Arena/)
29
+ Hugging Face: 04/09/2026 via [nvidia/Arena-Dexsuite-Lift-RL-Newton-Task](https://huggingface.co/nvidia/Arena-Dexsuite-Lift-RL-Newton-Task)
30
 
31
  ### Use Case:
32
  Researchers, Academics, Open-Source Community: AI-driven robotics research and algorithm development.
 
34
  Startups & Companies: Accelerate robotics development and reduce training costs.
35
 
36
  ## Reference(s):
37
+ [RSL‑RL: A Learning Library for Robotics Research, arXiv preprint (2025)](https://arxiv.org/abs/2509.10771)
38
 
39
  ## Model Architecture:
40
 
41
  **Architecture Type:** Multi-layer Perceptron
42
 
43
  **Network Architecture:**
44
+ MLP network trained with on-policy PPO using RSL-RL PPO framework.
45
 
46
  **Number of Model Parameters:**
47
  105,871
 
56
  **Input Parameters:**
57
  * Robot State: One-Dimensional (1D) - Floating number vector<br>
58
 
59
+ **Other Properties Related to Input:** Continuous-value vectors correspond to different motor controls on a robot, which depends on Degrees of Freedom of the robot embodiment.
60
+
61
  ## Output:
62
  **Output Type(s):** End-effector target pose<br>
63
  **Output Format:** Continuous-value vectors<br>
64
  **Output Parameters:** [One-Dimensional (1D)] <br>
65
+ **Other Properties Related to Output:** Continuous-value vectors correspond to different motor controls on a robot, which depends on Degrees of Freedom of the robot embodiment.
66
+
67
+ Our AI models are designed and/or optimized to run on NVIDIA GPU-accelerated systems. By leveraging NVIDIA's hardware (e.g. GPU cores) and software frameworks (e.g., CUDA libraries), the model achieves faster training and inference times compared to CPU-only solutions. <br>
68
 
69
+ The integration of foundation and fine-tuned models into AI systems requires additional testing using use-case-specific data to ensure safe and effective deployment. Following the V-model methodology, iterative testing and validation at both unit and system levels are essential to mitigate risks, meet technical and functional requirements, and ensure compliance with safety and ethical standards before deployment.
70
 
71
  ## Model Version(s):
72
  Version 0.1.
73
 
74
  # Training, Testing, and Evaluation:
 
75
 
76
+ ## Training & Testing
77
+
78
+ The model was trained completely in simulation (IsaacLab-Arena) using a on-policy Reinforcement Learning method.
79
+
80
+ **Data Modality:** Other: Robot simulation
81
+
82
+ **Training Data Size:** Not Applicable
83
+
84
+ **Data Collection Method:** Simulated Kuka Allegro - Reinforcement Learning
85
+
86
+ **Labeling Method:** Not Applicable
87
+
88
+ **Properties:** The model was trained completely in simulation (IsaacLab-Arena) using a on-policy Reinforcement Learning method. In this case, the data was collected for a couple of episodes, then the algorithm ran gradient descent on the collected data to update the model weights. Once this step was completed, the collected data was discarded and a new batch of data was collected.
89
+
90
+ The simulation environment contained the robot itself, a primitive sitting on a table, and a ground plane.
91
 
92
+ ## Evaluation
93
 
94
+ **Data Modality:** Other: Robot simulation
95
 
96
+ **Data Collection Method:** Simulated Kuka Allegro - Reinforcement Learning
97
 
98
+ **Labeling Method:** Not Applicable
99
 
100
+ **Properties:** The evaluation was performed in simulation using the IsaacLab-Arena (Newton physics engine). The evaluation data consists of dynamically generated episodes of lifting tasks.
101
 
102
  ## Inference:
103
  **Acceleration Engine(s):** PyTorch
104
 
105
  **Test Hardwares**
106
+ * NVIDIA RTX 6000 Ada
 
 
 
 
 
 
 
 
107
 
108
  ## Software Integration
109
 
 
119
 
120
  **Preferred/Supported Operating System(s):** Linux
121
 
 
 
122
  ## Ethical Considerations:
123
  NVIDIA believes Trustworthy AI is a shared responsibility and we have established policies and practices to enable development for a wide array of AI applications. When downloaded or used in accordance with our terms of service, developers should work with their internal model team to ensure this model meets requirements for the relevant industry and use case and addresses unforeseen product misuse.
124