Skip to content

Conversation

@mmcky
Copy link
Contributor

@mmcky mmcky commented Nov 12, 2025

This PR removes PyTorch and Pyro-PPL dependencies and switches from RunsOn GPU instances to GitHub-hosted runners with CPU-only JAX.

Changes:

  • Removed PyTorch and Pyro-PPL from all workflows (ci.yml, cache.yml, publish.yml)
  • Switched from runs-on GPU instances (g4dn.2xlarge) to GitHub-hosted runners (ubuntu-latest)
  • Changed JAX installation from CUDA variants to CPU-only: pip install -U jax
  • Removed nvidia-smi checks (no longer needed without GPU)
  • Updated step names to reflect CPU-only installation

This approach avoids CUDA driver compatibility issues by running JAX in CPU mode.

mmcky and others added 6 commits November 12, 2025 14:11
Create ~/.config/netlify directory with proper permissions before running
Netlify CLI to prevent EACCES errors. This issue occurs because the CLI
is installed globally with sudo but runs as the regular user, requiring
write access to the config directory.
Resolved conflicts in .github/workflows/ci.yml by keeping the CPU-only
JAX configuration and GitHub-hosted runner setup from this branch.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants