Project Info
Inspiration
At the age of nine years old, I taught myself to code. My school’s librarian introduced me to Scratch, an open-source platform that inevitably resulted in my journey in computer science. While platforms like Neetcode, Scratch, and Khan Academy teach the fundamentals of computer science, they don’t cover the details necessary to understand the workings of machine learning algorithms. We remedied this.
What it does
Burn allows users to build, train, test, and deploy custom machine learning models regardless of their level of experience. With tooling intuitive to both beginners and experienced engineers, anyone can use Burn to create custom (convolutional) neural networks, transformer-based models, or regression & classification models. Burn simplifies the model development process using a 3-dimensional sandbox experience, allowing users to physically engage with the systems they design. With support for both local (Mac, PC, DGX Spark, Jetson Nano, etc) and cloud (Modal) services, we enable any user to work in machine learning. Additionally, we allow the user to immerse themself into the learning process using a Cosmos-based (Vision Language Model) Tutor to explain exactly what’s going on in the development process.
How we built it
Connecting our Python (PyTorch, Scikit-learn, etc) backend with our React (TypeScript) frontend was Fast API. We served endpoints for the two to communicate, creating a seamless machine learning research & engineering experience with production-grade UI (inspired by Unity!).
Challenges we ran into
One major issue we encountered was leveraging Edge AI. After deploying our workloads to the Jetson Orin Nano, we found that even with its integrated GPU, it did not perform as well as running the workloads on Apple Silicon. This resulted in significant time loss, and although we completed the workflow for the Jetson, it was never deployed to production. Instead, we focused our efforts on the DGX Spark, which is compatible with all the models on Burn.
Accomplishments we're proud of
Initially, we planned to develop a completely different project. At first, our interested was in distributed inferring. However, we realized pretty early into the hackathon the underlying flaw in network-distributed learning as a whole and quickly pivoted to Burn. Our quick thinking allowed us to have enough time to still create an end-to-end, Tree Hacks-ready submission.
What we learned
Since our group consists of members with vastly different levels of experience, each of us learned something different from this project. One member, who is relatively new to working in a team-based software environment, learned the importance of communication and proper version control. Two members, who have more experience in traditional software engineering, primarily learned about machine learning, how it is developed, and the technology behind it. The final member, a machine learning engineer, gained a deeper understanding of the connection between front-end and back-end systems, as well as Three.js rendering.
What's next
We hope to continue Burn as a long-term open-source project to enable future generations to learn about artificial intelligence and machine learning, similar to platforms such as Khan Academy and Scratch. Authors Aryan Gupta Yax Patel Raj Pandya Benjamin Mujkic And a big shoutout to the organizers of Tree Hacks for having us for this great experience!
Burn
Build, Tweak, Deploy.
Burn enables people to create custom machine learning models using an intuitive sandbox UI. With visualizations to showcase concepts such as forward pass, gradient descent, backpropogation, and more in realtime, Burn helps users build an intuitive understanding of what's really happening under the hood. We currently support various models and algorithms such as Neural Networks (including AlexNet and an MNIST Database Classifier), Random Forest, Vision Language Models, and more.
See more on Devpost.
Latest: Burn is now Gradient.
Analysis
View
Metric
- 20
- 15
- 10
- 5
Figures cover GitHub contributors during the hackathon window. A co-authored commit counts in full for each author, so per-member totals add up to more than the whole-team figures.
Technology
- CSSIn code
- FastAPIIn code
- HTMLIn code
- Hugging FaceIn code
- JavaScriptIn code
- PythonIn code
- PyTorchIn code
- ReactIn code
- TypeScriptIn code
9 of 9 appear in the indexed code.
AI coding agents
No AI coding agent signals were found in this repository.
Detected from committed agent config files and commit authorship. Absence of a signal is not proof an agent was unused.
Codebase size
Source size
1.3 MB
Source files
160
Counts recognized source files only; vendored directories, binaries and lockfiles are excluded, so this is smaller than the repository on disk.
Repository
aryan-cs/burn
177 files · 1.9 MB · @ a6b5fe1
Structure
Interface
29 files · 16%Screens, components and styles rendered to the user.
Application logic
104 files · 59%Domain rules, services and shared utilities.
+1 moreData & schema
6 files · 3%Schema definitions, migrations and data access.
Supporting
Layers are inferred from where files sit in the tree, not from reading the code. A project that names its directories unconventionally will read oddly here — open the file browser to check anything the diagram implies.
Languages
- TypeScript57%
- Python34%
- CSS8%
- Markdown1%
- JavaScript0%
- HTML0%
Share of indexed source by file size. Binary and vendored files are excluded.
Dependencies
backend/pyproject.toml
pypi · 21- fastapi
- httpx
- ipython
- joblib
- kaggle
- matplotlib
- modal
- numpy
- pandas
- pillow
- pydantic
- pygame
- python-dotenv
- python-multipart
- scikit-learn
- tinygrad
- torch
- torchvision
- +3 more
frontend/package.json
npm · 19- @react-three/drei
- @react-three/fiber
- react
- react-dom
- three
- zustand
- +13 more
compute_node/pyproject.toml
pypi · 8- fastapi
- httpx
- pillow
- pydantic
- torch
- torchvision
- transformers
- uvicorn
Declared in the repository’s manifests at the indexed commit. A declared package is not proof it is used, and runtime dependencies are listed first.
This project’s features have not been analysed yet.
Export this project's context (description, README, evidence, key source files) to chat with an AI agent elsewhere.