The Super Bowl champ danced and chatted with friends following a weekend with girlfriend Taylor Swift at the festival

Travis Kelce spotted during Coachella. PHOTO: PEOPLE

Travis Kelce is “pumped up”  — and closing down Coachella!

The Super Bowl champ, 34, was seen partying with friends at Desert Nights presented by Tao Group Hospitality with CMG and PATRÓN El ALTO on Sunday, April 14, capping a festive weekend in the desert along with girlfriend Taylor Swift.

Kelce, who was seen at the bash Sunday night without Swift, wore a backwards green New Heights hat and sweatshirt in a video taken from the festival. His table was treated to bottle service as a line of servers carried light-up displays of Patrón El Cielo to their closed-off VIP table.

A source tells PEOPLE that the athlete was hanging with friends, laughing as they took shots and was dancing a lot when Adriatique, the last DJ of the weekend, came on. Per the video, the Kansas City Chiefs tight end was also seen dancing in place and smiling widely as Foster the People’s “Pumped Up Kicks” played in the background.

Earlier in the weekend, Kelce and the “Karma” singer, 34, were spotted in the crowd —  and behind the scenes — supporting famous friends at the festival.

The couple danced and embraced in the crowd during sets by Ice Spice, Dom Dolla and Jungle. They were also seen kissing during Jack Antonoff’s Bleachers performance and hanging out with Sabrina Carpenter, who opened for some of Swift’s Eras Tour dates, and actor Barry Keoghan.

While at the music festival outside Palm Springs, California, the couple embraced casual looks. The Grammy winner donned Kelce’s New Heights cap, which rounded out her all-black look featuring a tank top, pleated miniskirt with built-in garter, oversized leather blazer and sneakers.

JAMES DEVANEY/GC IMAGES, CHRISTOPHER POLK/VARIETY VIA GETTY, KEVIN WINTER/GETTY

Meanwhile, the NFL star channeled the festival style by sporting an open plaid shirt over a white T-shirt with striped pants and white Chucks.

The couple also coordinated their appearance with Stella McCartney accessories. Swift wore a shoulder bag with the brand’s logo across the strap while Kelce opted for oversize square sunglasses.

Prior to heading to Coachella, the couple was photographed leaving a sushi date night in Los Angeles on Friday, April 12. For the low-key dinner, the singer wore a red and white Kansas City Chiefs-esque jacket over a dark top and denim skirt.

Swift and Kelce have been keeping a relatively low profile while the superstar is on break from her Eras Tour and the football star is gearing up to film Are You Smarter Than a Celebrity.