Introducing the Trainwave Jupyter Extension
September 19, 2025
โข
Johan Backman

The Trainwave Jupyter Extension revolutionizes how data scientists and ML engineers work with cloud GPU resources. Transform your notebooks into scalable GPU jobs instantly, monitor progress in real-time, and access results without ever leaving your development environment.
The Problem with Traditional GPU Computing Workflows
If you've ever worked with cloud GPU resources for machine learning, you know the pain points:
- Context Switching: Constantly jumping between your notebook, terminal, and web browser
- Manual File Management: Uploading notebooks, managing dependencies, and tracking outputs across different platforms
- Complex Setup: Configuring environments, managing authentication, and dealing with CLI commands
- Disconnected Workflow: Losing the seamless experience of interactive development
These friction points slow down experimentation and make it harder to iterate quickly on your ML models.
Introducing the Trainwave Jupyter Extension
The Trainwave Jupyter Extension eliminates these pain points by bringing cloud GPU computing directly into your JupyterLab environment. Here's what makes it special:
๐ One-Click Job Launch
Convert any notebook into a GPU job with a single click. The extension automatically saves your notebook and submits it to Trainwave's powerful GPU infrastructure.
๐ Seamless Authentication
Integrated login with your Trainwave.ai account means no more managing API keys or tokens manually. Sign in once and you're ready to go.
๐ Real-time Monitoring
Track your job's progress directly from the extension. See status updates, GPU utilization, and completion notifications without leaving your notebook.
โ๏ธ Flexible Configuration
Choose your GPU type, specify the number of GPUs, and configure project settings - all through an intuitive interface that feels native to JupyterLab.
๐ฑ Modern, Integrated UI
The extension features a clean, modern interface that integrates seamlessly with JupyterLab's design language. It feels like a natural part of your development environment.
Key Features at a Glance
- One-Click Job Launch: Convert notebooks to GPU jobs directly from the toolbar
- Secure Authentication: Integrated login with your Trainwave.ai account
- Real-time Job Monitoring: Track job status and progress in real-time
- Flexible Configuration: Customize GPU types, counts, and project settings
- Modern UI: Clean, intuitive interface that integrates seamlessly with JupyterLab
- Auto-save: Automatically saves your notebook before launching jobs
- Job History: View and manage your recent jobs from the extension
Getting Started
Installation
Installing the extension is straightforward:
pip install trainwave-jupyter
After installation, restart JupyterLab and you'll see the Trainwave icon in your notebook toolbar.
Quick Setup
- Sign In: Click the Trainwave icon and authenticate with your Trainwave.ai account
- Configure Settings: Select your organization, project, and preferred GPU configuration
- Launch Jobs: Open any notebook and click "Launch Job" to run it on GPU infrastructure
Example Workflow
Here's how easy it is to use:
- Develop Locally: Write and test your ML code in your Jupyter notebook
- Launch to GPU: Click the Trainwave icon and select "Launch Job"
- Monitor Progress: Watch real-time updates on your job's progress
- Access Results: View outputs and artifacts directly in the Trainwave web interface
Perfect for Data Scientists and ML Engineers
The extension is designed specifically for the workflows that data scientists and ML engineers use every day:
- Rapid Prototyping: Test ideas quickly on local resources, then scale up to GPUs when needed
- Interactive Development: Maintain the interactive nature of Jupyter while leveraging cloud resources
- Experiment Management: Keep track of different experiments and their results
- Collaborative Work: Share notebooks and results easily with team members
Technical Details
- Compatibility: Works with JupyterLab 4.0+ and Python 3.9+
- GPU Support: Access to a wide range of GPU types including T4, V100, A100, and more
- Real-time Updates: Automatic polling for job status updates
- Secure: All communication is encrypted and authenticated
Try It Today
The Trainwave Jupyter Extension is now available on PyPI. Whether you're training deep learning models, running data analysis on large datasets, or experimenting with new ML techniques, this extension will streamline your workflow and make cloud GPU computing feel as natural as running code locally.
Ready to transform your Jupyter workflow? Install the extension today and experience the future of interactive ML development.
Have questions or feedback about the extension? We'd love to hear from you! Join our community or reach out to our support team.