robotic

Equivariant DQN in CartPole

A project that implements Deep Q-Networks (DQN) in OpenAI Gym's CartPole environment, employing equivariant model techniques to achieve faster training speeds.

Grasping Model with Enhanced SE(2) Prediction

This project enhances a robotic grasping model using PyTorch by integrating safe-z prediction into an SE(2) model, achieving faster training and optimization for patch-based input.