Graphzero
Zero-copy C++ graph engine to train PyTorch GNNs with 0 RAM.
What it does
GraphZero fixes PyTorch Geometric OOM crashes by memory-mapping massive graph datasets directly from your SSD. Built with C++20 and nanobind, it hands raw pointers to PyTorch as zero-copy NumPy arrays. Train 50GB models on consumer hardware by letting the OS handle page faults while the GPU focuses entirely on the math.
Does the same job
all alternatives →- ISI stripped DALL·E Mini to its bare essentials and converted it to Torch2022 · github.com · ▲663
- LFLeanRL: Fast PyTorch RL with Torch.compile and CUDA Graphs2024 · github.com · ▲53
We're excited to announce that we've open-sourced LeanRL, a lightweight PyTorch reinforcement learning library that provides recipes for fast RL training using torch.compile and CUDA graphs. By leveraging these tools, we've achieved significant speed-ups compared to the original CleanRL implementations - up to 6x faster! Reinforcement learning is notoriously CPU-bound due to the high frequency of small CPU operations. PyTorch's powerful compiler can help alleviate these issues, but comes with its own costs. LeanRL addresses this challenge by providing simple recipes to accelerate your…
- KYKeep your PyTorch model in VRAM by hot swapping code2025 · github.com · ▲77
- TLTorch Lens Maker – Differentiable Geometric Optics in PyTorch2025 · victorpoughon.github.io · ▲179
Hello HN! For the past 6 months I've been working on an open source python library that implements differentiable geometric optics in PyTorch. It's very experimental still, but eventually the goal is to use it to design optical systems with a state of the art optimization framework and a beautiful code based API. Think OpenSCAD, but for optical systems. Not only is PyTorch's autograd an amazing general purpose optimizer, but torch.nn (the neural network building blocks) can be used pretty much out of the box to model an optical system. This is because there is a strong analogy to be made…
- GFGraphlearn-for-PyTorch, distributed graph learning on PyTorch2023 · github.com · ▲9
Hello HN, We are pleased to introduce you graphlearn-for-pytorch (https://github.com/alibaba/graphlearn-for-pytorch), an open-source distributed graph neural network library based on PyTorch and compatible with PyG. Our library is designed to make it easy for developers to build and train large-scale graph models in a distributed environment. With graphlearn-for-pytorch, you can leverage GPUs to accelerate graph sampling and utilize UVA to reduce the overheads of feature collection. Following a scalable design, graphlearn-for-pytorch supports training GNN models on…
More ai this month
the category →
I trained a 125M-parameter transformer to autocomplete piano performances in real time (~108 notes/sec on an iPhone 15). The idea is basically GitHub Copilot or Tabnine, except instead of prompting it with code, you prompt it by playing a few notes on a MIDI piano. The model then continues what you played, entirely on-device. The app is free if anyone wants to try it. Happy to answer questions about the model, training, Core ML, or the many things that didn't work.
AI · 17d ago · simedw.com
Astute▲585Automate your B2B brand going viral, with new media creators
AI · 18d ago · company-app.joinastute.com
Hey HN, Henry from Cactus here! We previously released Cactus Needle, a 14MB agentic LLM for tool call, device use, and structured extraction for phones, wearables, smart homes, small robots and microcontrollers. We got really great feedback here, and have now incorporated the suggestions to release Needle 2. The whole model is a single 14MB binary that runs a full session in 28MB of RAM; 45m parameters at 2bit compression. Needle hits 500 tokens/sec decode speed on a Raspberry Pi 5, sits between 400-1,500 tokens/sec on VR devices like Meta Quest 3S and Apple Vision Pro, and ranges…
AI · 27d ago · cactuscompute.com
Launched alongside, March 2026
the whole month →
Switch from ChatGPT to Claude with import memory feature
AI · Mar 2026 · claude.com
