JAX series part 2: JAX for Combinatorial Optimisation

GDG AI for Science - Australia

Dive into parallel continuous local search for SAT and pseudo-Boolean problems. Learn how JAX enables high-performance, scalable search algorithms to tackle complex problems efficiently. Discover the power of GPU-accelerated optimisation.

Jul 4, 1:00 – 2:00 AM (UTC)

30 RSVP'd

RSVP

Key Themes

AIJAX

About this event

We investigate parallel search algorithms for the Satisfiability (SAT) problem—with Boolean SAT widely recognised as the first and canonical NP-Complete problem. Such algorithms have broad applications across domains such as computer security, cryptology, bioinformatics, spectrum auction design, formal verification, and automated planning. We will briefly motivate and describe a class of pseudo-Boolean SAT problems.

SAT problems are typically approached using two primary search paradigms: systematic search and stochastic local search. In this talk, we focus on a contemporary variant of the latter—continuous local search—an approach amenable to execution and acceleration via parallelisation on GPUs. We will discuss how to leverage JAX - a high-performance numerical computing library-to accelerate and scale key components of continuous local search using GPU parallelism. This will show how JAX can be used to efficiently implement and experiment with parallel search algorithms for evaluating SAT.

Speakers

  • Cody Christopher

    Australian National University

    Research Fellow

  • Charles Gretton

    Australian National University

    Associate Professor

Organizers

  • Susan Wei

    Monash University

    Organizer

  • Pablo Rozas Larraondo

    Haizea Analytics

    Organizer

  • Marcel Scharth

    The University of Sydney

    Organizer

  • Lifi Huang

    Monash University

    Organizer

  • Mauricio Marrone

    Macquarie University

    Macquarie University

  • David Kainer

    The University of Queensland

    University of Queensland

  • Nathaniel Butterworth

    Google

    Science Catalyst Program Manager

Contact Us