CEREBRAS ADVANCED TECHNOLOGY · RESEARCH PREVIEW
CEREBRAS ADVANCED TECHNOLOGY · RESEARCH PREVIEWExperimental SDK
The Cerebras Experimental Software Development Kit provides high-level languages and tools for researchers and developers writing scientific and high-performance kernels for the Cerebras Wafer-Scale Engine.
TUNGSTEN + PAINT
Tutorial
The tutorial teaches both languages one idea at a time, from a single tile to programs that span many. Each chapter builds and simulates on its own. It is ready for coding agents: with the tutorial and the language references, yours can write and test a new kernel.
Hello, wafer!
The smallest complete program: one PE adds 1 to each element of an array. How Tungsten, Paint and Carbide fit together, and how a test builds, runs and checks it.
Open lesson 04Sockets, wavelets, colors and routes
A stream of numbers through four PEs: sockets in Tungsten, colors and routes in Paint, and one stream delivered to two receivers.
Open lesson 08Space and time: a wavefront along a row
One kernel on a whole row of PEs aligns two DNA sequences: each PE owns a column of the score matrix, and the rows stream through as a wavefront.
Open lesson 11A grid of blocks: halo exchange and what it costs
A Jacobi stencil on a 2-D field, one block per PE: every step, each PE trades its block edges with its four neighbours, and the lesson measures that exchange against the computation.
Open lessonSCIENTIFIC CODES
Applications
Each application is a complete program with all of its source files, a page that walks through its algorithm, data exchange and routing, and a test that builds it, simulates it and checks the result against a NumPy reference.
Browse every project in the Aviary- HEAT9Heat equationHeat flow; the canonical stencil codeEvolve a two-dimensional heat field with a nine-point stencil.
- GSGray-Scott reaction-diffusionPattern formation in chemistry and biologyEvolve two reacting, diffusing chemicals with a nine-point Laplacian, one tile of grid cells per PE.
- SWEShallow-water equationsTides, storm surges and tsunamis in coastal basinsPropagate linear gravity waves on a rotating plane in a closed basin, a block of grid cells per tile.
- MDShort-range molecular dynamicsAtom-by-atom simulation of liquids and materialsExchange particles and evaluate short-range pair interactions across a spatial decomposition.
- EAMEmbedded-atom simulationMetals and alloys under stress, heat and impactEvaluate tabulated embedded-atom interactions with distributed particle exchange.
- SNAPParticle transport sweepsNeutron and radiation transport in reactors and shieldingSweep angular fluxes across a 2D grid, one cell per PE, with SNAP's discrete ordinates method.
- FFTFast Fourier TransformSpectral methods and signal processingCompute the discrete Fourier transform using a distributed algorithm.
- HPCGHigh-Performance Conjugate GradientsThe HPCG supercomputer benchmarkSolve a 27-point sparse linear system by conjugate gradient with a multigrid preconditioner, one grid column per PE, in single precision (32-bit).
- PICElectromagnetic particle-in-cellElectron beams in laboratory plasmasAdvance charged particles and the electromagnetic fields they drive, one mesh cell per PE, with an implicit Maxwell solver.
- HOMHigh-order hydrodynamicsCompressible flow on a mesh that moves with the materialMove a two-dimensional compressible gas and its mesh together with high-order finite elements: matrix-free mass and force actions, a conjugate gradient solve and two-stage Runge-Kutta steps.