A CUDA-accelerated implementation of the GRPO algorithm for efficient model training.
Clone the repository and navigate to the project directory:
git clone https://github.com/Hajiyevrehman/grpo-cuda.git
cd grpo-cudaThe project requires several dependencies:
# Install Python packages
pip install pyngrok
# Install system dependencies
sudo apt-get update
sudo apt-get install -y ninja-buildThis project includes an API server that can be exposed via ngrok for remote access:
from pyngrok import ngrok
# Set your ngrok authentication token
# You can get a token by signing up at https://ngrok.com/
ngrok.set_auth_token("YOUR_NGROK_AUTH_TOKEN")
# Start the tunnel on port 5000
public_url = ngrok.connect(5000)
print(f"Your API is accessible at: {public_url.public_url}/test_kernel")Make the start script executable and run it:
chmod +x start_server.sh
./start_server.shOnce the server is running, you can access the API at the ngrok URL printed to the console. The API endpoint is available at /test_kernel.
- If you encounter permission issues when running apt commands, try using
sudoinstead. - Make sure your ngrok authentication token is valid.
- Check system requirements for CUDA compatibility.
[Add license information here]
[Add contributor information here]