Merge branch 'main' into feature/get-weld

Author: LeonLiu4

.github/workflows/generic.yml

@@ -16,7 +16,7 @@ jobs:
       matrix:
         # See official Github documentation for details: https://shorturl.at/NJgsj
         OS: ["ubuntu-22.04", "ubuntu-24.04", "macos-15", "windows-cpu-4-core"]
-        PYTHON_VERSION: ["3.10", "3.11", "3.12"]
+        PYTHON_VERSION: ["3.10", "3.11", "3.12", "3.13"]
 
     env:
       HF_HUB_DOWNLOAD_TIMEOUT: 60
@@ -103,7 +103,7 @@ jobs:
 
       - name: Install Genesis
         run: |
-          pip install -e '.[dev,render]'
+          pip install -e '.[dev]'
 
       - name: Run unit tests
         run: |

.github/workflows/production.yml

@@ -22,8 +22,9 @@ jobs:
       WANDB_API_KEY: ${{ secrets.WANDB_API_KEY }}
       HF_TOKEN: ${{ secrets.HF_TOKEN }}
       HF_HUB_DOWNLOAD_TIMEOUT: 60
-      GENESIS_IMAGE_VER: "1_0"
+      GENESIS_IMAGE_VER: "1_1"
       TIMEOUT_MINUTES: 180
+      MADRONA_DISABLE_CUDA_HEAP_SIZE: "1"
 
     steps:
       - name: Checkout code

README.md

@@ -77,7 +77,7 @@ Install **PyTorch** first following the [official instructions](https://pytorch.
 
 Then, install Genesis via PyPI:
 ```bash
-pip install genesis-world  # Requires Python>=3.10,<3.13;
+pip install genesis-world  # Requires Python>=3.10,<3.14;
 ```
 
 For the latest version to date, make sure that `pip` is up-to-date via `pip install --upgrade pip`, then run command:
@@ -172,6 +172,7 @@ Genesis's development has been made possible thanks to these open-source project
 - [libccd](https://github.com/danfis/libccd): Reference for collision detection.
 - [PyRender](https://github.com/mmatl/pyrender): Rasterization-based renderer.
 - [LuisaCompute](https://github.com/LuisaGroup/LuisaCompute) and [LuisaRender](https://github.com/LuisaGroup/LuisaRender): Ray-tracing DSL.
+- [Madrona](https://github.com/shacklettbp/madrona) and [Madrona-mjx](https://github.com/shacklettbp/madrona_mjx): Batch renderer backend
 
 ## Associated Papers
 

README_FR.md

@@ -69,7 +69,7 @@ Page du projet : <https://genesis-embodied-ai.github.io/>
 Genesis est disponible via PyPI :
 
 ```bash
-pip install genesis-world  # Nécessite Python>=3.10,<3.13;
+pip install genesis-world  # Nécessite Python>=3.10,<3.14;
 ```
 
 Vous devez également installer **PyTorch** en suivant [les instructions officielles](https://pytorch.org/get-started/locally/).

README_JA.md

@@ -69,7 +69,7 @@ Genesisの目指すところ：
 GenesisはPyPIで利用可能です：
 
 ```bash
-pip install genesis-world  # Python>=3.10,<3.13 が必要です;
+pip install genesis-world  # Python>=3.10,<3.14 が必要です;
 ```
 
 また、**PyTorch**を[公式手順](https://pytorch.org/get-started/locally/)に従ってインストールする必要があります。

README_KR.md

@@ -69,7 +69,7 @@ Genesis의 목표:
 Genesis는 PyPI를 통해 설치할 수 있습니다:
 
 ```bash
-pip install genesis-world  # Python>=3.10,<3.13 필요
+pip install genesis-world  # Python>=3.10,<3.14 필요
 ```
 
 또한, [공식 설명서](https://pytorch.org/get-started/locally/)에 따라 **PyTorch**를 설치해야 합니다.

examples/collision/contype.py

@@ -0,0 +1,89 @@
+"""
+NOTE: contype and conaffinity are 32-bit integer bitmasks used for contact filtering of contact pairs.
+When the contype of one geom and the conaffinity of the other geom share a common bit set to 1, two geoms can collide.
+Plane:      contype=0xFFFF, conaffinity=0xFFFF (1111 1111 1111 1111)
+Red Cube:   contype=1, conaffinity=1 (0001) -> collide with Plane and Blue Cube
+Green Cube: contype=2, conaffinity=2 (0010) -> collide with Plane and Blue Cube
+Blue Cube:  contype=3, conaffinity=3 (0011) -> collide with Plane, Red Cube, and Green Cube
+Dragon:     contype=4, conaffinity=4 (0100) -> collide with Plane only
+"""
+
+import argparse
+
+import genesis as gs
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("-v", "--vis", action="store_true", default=False)
+    args = parser.parse_args()
+
+    gs.init()
+
+    scene = gs.Scene(
+        viewer_options=gs.options.ViewerOptions(
+            camera_pos=(0.0, -2, 1.5),
+            camera_lookat=(0.0, 0.0, 0.5),
+            camera_fov=40,
+            max_FPS=200,
+        ),
+        show_viewer=args.vis,
+    )
+
+    scene.add_entity(gs.morphs.Plane())
+
+    scene.add_entity(
+        gs.morphs.Box(
+            pos=(0.025, 0, 0.5),
+            quat=(0, 0, 0, 1),
+            size=(0.1, 0.1, 0.1),
+            contype=1,
+            conaffinity=1,
+        ),
+        surface=gs.surfaces.Default(
+            color=(1.0, 0.0, 0.0, 1.0),
+        ),
+    )
+    scene.add_entity(
+        gs.morphs.Box(
+            pos=(-0.025, 0, 1.0),
+            quat=(0, 0, 0, 1),
+            size=(0.1, 0.1, 0.1),
+            contype=2,
+            conaffinity=2,
+        ),
+        surface=gs.surfaces.Default(
+            color=(0.0, 1.0, 0.0, 1.0),
+        ),
+    )
+    scene.add_entity(
+        gs.morphs.Box(
+            pos=(0.0, 0, 1.5),
+            quat=(0, 0, 0, 1),
+            size=(0.1, 0.1, 0.1),
+            contype=3,
+            conaffinity=3,
+        ),
+        surface=gs.surfaces.Default(
+            color=(0.0, 0.0, 1.0, 1.0),
+        ),
+    )
+    scene.add_entity(
+        morph=gs.morphs.Mesh(
+            file="meshes/dragon/dragon.obj",
+            scale=0.004,
+            euler=(0, 0, 90),
+            pos=(-0.1, 0.0, 1.0),
+            contype=4,
+            conaffinity=4,
+        ),
+    )
+
+    scene.build()
+
+    for i in range(1000):
+        scene.step()
+
+
+if __name__ == "__main__":
+    main()

examples/coupling/fem_cube_linked_with_arm.py

@@ -130,7 +130,7 @@ def main():
                 cam.render()
 
         print("Now dropping the cube")
-        cube.solver.remove_vertex_constraints()
+        cube.remove_vertex_constraints()
         steps = int(1 / dt)
         for i in tqdm(range(steps), total=steps):
             arm.control_dofs_position(arm_path_waypoints[-1])

examples/drone/hover_env.py

@@ -2,7 +2,12 @@
 import math
 import copy
 import genesis as gs
-from genesis.utils.geom import quat_to_xyz, transform_by_quat, inv_quat, transform_quat_by_quat
+from genesis.utils.geom import (
+    quat_to_xyz,
+    transform_by_quat,
+    inv_quat,
+    transform_quat_by_quat,
+)
 
 
 def gs_rand_float(lower, upper, shape, device):
@@ -147,7 +152,10 @@ def step(self, actions):
         self.last_rel_pos = self.commands - self.last_base_pos
         self.base_quat[:] = self.drone.get_quat()
         self.base_euler = quat_to_xyz(
-            transform_quat_by_quat(torch.ones_like(self.base_quat) * self.inv_base_init_quat, self.base_quat),
+            transform_quat_by_quat(
+                torch.ones_like(self.base_quat) * self.inv_base_init_quat,
+                self.base_quat,
+            ),
             rpy=True,
             degrees=True,
         )

examples/drone/interactive_drone.py

@@ -1,16 +1,17 @@
-import argparse
-import numpy as np
-import genesis as gs
 import time
 import threading
+
 from pynput import keyboard
+import numpy as np
+
+import genesis as gs
 
 
 class DroneController:
     def __init__(self):
-        self.thrust = 14468.429183500699  # Base hover RPM - constant hover
-        self.rotation_delta = 200  # Differential RPM for rotation
-        self.thrust_delta = 10  # Amount to change thrust by when accelerating/decelerating
+        self.thrust = 14475.8  # Base hover RPM - constant hover
+        self.rotation_delta = 200.0  # Differential RPM for rotation
+        self.thrust_delta = 10.0  # Amount to change thrust by when accelerating/decelerating
         self.running = True
         self.rpms = [self.thrust] * 4
         self.pressed_keys = set()
@@ -93,46 +94,37 @@ def update_thrust(self):
 
 def run_sim(scene, drone, controller):
     while controller.running:
-        try:
-            # Update drone with current RPMs
-            rpms = controller.update_thrust()
-            drone.set_propellels_rpm(rpms)
+        # Update drone with current RPMs
+        rpms = controller.update_thrust()
+        drone.set_propellels_rpm(rpms)
 
-            # Update physics
-            scene.step()
+        # Update physics
+        scene.step(refresh_visualizer=False)
 
-            time.sleep(1 / 60)  # Limit simulation rate
-        except Exception as e:
-            print(f"Error in simulation loop: {e}")
-
-    if scene.viewer:
-        scene.viewer.stop()
+        # Limit simulation rate
+        time.sleep(1.0 / scene.viewer.max_FPS)
 
 
 def main():
-    parser = argparse.ArgumentParser()
-    parser.add_argument("-v", "--vis", action="store_true", default=True, help="Enable visualization (default: True)")
-    parser.add_argument("-m", "--mac", action="store_true", default=False, help="Running on MacOS (default: False)")
-    args = parser.parse_args()
-
     # Initialize Genesis
     gs.init(backend=gs.cpu)
 
-    # Create scene with initial camera view
-    viewer_options = gs.options.ViewerOptions(
-        camera_pos=(0.0, -4.0, 2.0),  # Now behind the drone (negative Y)
-        camera_lookat=(0.0, 0.0, 0.5),
-        camera_fov=45,
-        max_FPS=60,
-    )
-
+    # Create scene
     scene = gs.Scene(
         sim_options=gs.options.SimOptions(
             dt=0.01,
             gravity=(0, 0, -9.81),
         ),
-        viewer_options=viewer_options,
-        show_viewer=args.vis,
+        viewer_options=gs.options.ViewerOptions(
+            camera_pos=(0.0, -2.0, 1.0),
+            camera_lookat=(0.0, 0.0, 0.3),
+            camera_fov=45,
+            max_FPS=60,
+        ),
+        vis_options=gs.options.VisOptions(
+            show_world_frame=False,
+        ),
+        show_viewer=True,
     )
 
     # Add entities
@@ -146,38 +138,32 @@ def main():
 
     scene.viewer.follow_entity(drone)
 
-    # Build scene
-    scene.build()
-
     # Initialize controller
     controller = DroneController()
 
+    # Start keyboard listener.
+    # Note that instantiating the listener after building the scene causes segfault on MacOS.
+    listener = keyboard.Listener(on_press=controller.on_press, on_release=controller.on_release)
+    listener.start()
+
+    # Build scene
+    scene.build()
+
     # Print control instructions
     print("\nDrone Controls:")
     print("↑ - Move Forward (North)")
     print("↓ - Move Backward (South)")
     print("← - Move Left (West)")
     print("→ - Move Right (East)")
+    print("space - Increase RPM")
+    print("shift - Decrease RPM")
     print("ESC - Quit\n")
     print("Initial hover RPM:", controller.thrust)
 
-    # Start keyboard listener
-    listener = keyboard.Listener(on_press=controller.on_press, on_release=controller.on_release)
-    listener.start()
-
-    if args.mac:
-        # Run simulation in another thread
-        sim_thread = threading.Thread(target=run_sim, args=(scene, drone, controller))
-        sim_thread.start()
-
-        if args.vis:
-            scene.viewer.start()
+    # Run simulation in another thread
+    threading.Thread(target=run_sim, args=(scene, drone, controller)).start()
+    scene.viewer.run()
 
-        # Wait for threads to finish
-        sim_thread.join()
-    else:
-        # Run simulation in main thread
-        run_sim(scene, drone, controller)
     listener.stop()