TPU7x (Ironwood) supported configurations

The slice shapes you can request for TPU7x (Ironwood), and how many chips and hosts each uses.

Slice shapes

Note: A slice shape gives the number of chips along each dimension of the interconnect. A 2D shape is written AxB and a 3D shape AxBxC. Multiply the dimensions to get the number of chips in the slice.

Shape

Chips

Hosts

VMs

Cubes

Scope

2x2x1

4

1

1

1/16

Single-host

2x2x2

8

2

2

1/8

Multi-host

2x2x4

16

4

4

1/4

Multi-host

2x4x4

32

8

8

1/2

Multi-host

4x4x4

64

16

16

1

Multi-host

4x4x8

128

32

32

2

Multi-host

4x8x8

256

64

64

4

Multi-host

8x8x8

512

128

128

8

Multi-host

8x8x16

1024

256

256

16

Multi-host

8x16x16

2048

512

512

32

Multi-host

Shapes also follow the general form AxBxC, where each dimension is a multiple of 4, up to 9216 chips. Every shape uses the tpu7x-standard-4t machine type. Slices larger than 2048 chips require contacting your account team.

Parent topic:TPU7x (Ironwood)