Commit Graph

9 Commits

Author SHA1 Message Date
google-labs-jules[bot]
2781d216c6 feat: Refactor nanochat to run on Vertex AI Pipelines
This refactoring enables the nanochat project to be executed as a scalable and robust pipeline on Vertex AI.

The monolithic `speedrun.sh` script has been decomposed into a series of containerized components orchestrated by a Kubeflow pipeline.

The codebase has been updated to use Google Cloud Storage for artifact management, allowing for seamless data sharing between pipeline steps.

A `Dockerfile` and Python wrappers for each pipeline step have been added to the `vertex_pipelines` directory.

The `README.md` has been updated with instructions on how to build the Docker image and run the Vertex AI pipeline.
2025-11-04 01:26:51 +00:00
Ajeesh Sunil
5e0987a431 numpy isnt acting as a dependency for nanochat, so isnt it better to remove numpy from dependencies list 2025-10-28 20:05:38 +00:00
Luke Stanley
901b075605 Fix GPU-less CPU use on Linux with specific Torch indexes 2025-10-21 23:14:16 +00:00
burtenshaw
0abb0fa2e3 add both sides of the source check 2025-10-20 10:44:07 +02:00
burtenshaw
c7ae920a77 add check for linux on cpu 2025-10-20 06:51:52 +02:00
karpathy
e4f9b9c64d revert to previous pyproject.toml 2025-10-17 08:08:16 -07:00
burtenshaw
23b6351c1c add groups and source selection 2025-10-17 12:20:18 +02:00
karpathy
306bc380ab add support for CPU and for MPS. I had to change a few cosmetic things. I also discovered I think a bit of a bug, where I was casting wte to bfloat16 in the wrong place (the model init) instead of in init_weights 2025-10-16 10:04:43 -07:00
karpathy
3a5e0bc50b initial commit 2025-10-13 06:49:24 -07:00