Skip to content

@uwplasma UW Plasma

All

    Repositories list

    49 repositories

    • ESSOS

      Public
      e-Stellarator Simulation and Optimization Suite
      optimizationautogradfusion
      optimizationautogradfusionjax
      Python
      •
      MIT License
      •5•19•4•36•Updated Oct 27, 2026Oct 27, 2026

    • Dark-SPECTRAX

      Public
      Kinetic Maxwell-Proca (dark photon) simulations with Cartesian Hermite-Fourier discretization, built on SPECTRAX
      Python
      •
      Other
      •0•0•0•0•Updated Oct 27, 2026Oct 27, 2026

    • DKX

      Public
      JAX Version of SFINCS
      Python
      •
      MIT License
      •2•2•0•13•Updated Oct 27, 2026Oct 27, 2026

    • vmex

      Public
      JAX Version of VMEC2000
      Python
      •
      MIT License
      •4•16•1•21•Updated Oct 26, 2026Oct 26, 2026

    • GKX

      Public
      Gyrokinetic Spectral Solver of Plasma Turbulence
      Python
      •
      MIT License
      •1•6•0•4•Updated Oct 26, 2026Oct 26, 2026

    • LMhdX

      Public
      LaMinar inductionless MHD code in JAX
      Python
      •
      MIT License
      •0•2•1•20•Updated Oct 26, 2026Oct 26, 2026

    • DRBX

      Public
      Drift-Reduced Braginskii Solver in JAX
      Python
      •
      MIT License
      •0•4•5•5•Updated Oct 26, 2026Oct 26, 2026

    • SPECTRAX

      Public
      Solves Vlasov-Maxwell equations by doing a Hermite-Fourier decomposition.
      Python
      •
      MIT License
      •2•5•1•37•Updated Oct 26, 2026Oct 26, 2026

    • NEOPAX

      Public
      NEOPAX: Neoclassical Transport Package in JAX
      Python
      •
      MIT License
      •1•4•4•4•Updated Oct 26, 2026Oct 26, 2026

    • NTX

      Public
      Neoclassical Transport Code
      Python
      •
      MIT License
      •0•1•0•11•Updated Oct 26, 2026Oct 26, 2026

    • SOLVAX

      Public
      Differentiable structured linear solvers, preconditioners and matrix-free methods in JAX
      Python
      •
      MIT License
      •2•6•4•1•Updated Oct 25, 2026Oct 25, 2026

    • fax

      Public
      Differentiable JAX code for Alfvén eigenmodes and fast-ion physics in stellarators
      Python
      •
      MIT License
      •0•0•0•4•Updated Oct 25, 2026Oct 25, 2026

    • NEO_JAX

      Public
      Calculation of epsilon effective using a JAX version of NEO from STELLOPT
      Python
      •
      MIT License
      •0•2•0•3•Updated Oct 24, 2026Oct 24, 2026

    • PyPIC3D

      Public
      PyPIC3D is a 3D particle-in-cell code written in Python using Jax.
      particle-in-cellplasma-physics3d-simulation
      particle-in-cellplasma-physics3d-simulationjaxautodifferentiationmadison-wisconsin
      Python
      •
      MIT License
      •3•4•0•0•Updated Oct 22, 2026Oct 22, 2026

    • JAX_BSSN

      Public
      JAX BSSN is an implementation of the BSSN evolution equations for fully dynamical simulations of the Einstein field equations
      scientific-computingnumerical-relativityautodiff
      scientific-computingnumerical-relativityautodiffjaxbssneinstein-field-equations
      Python
      •1•1•0•0•Updated Oct 22, 2026Oct 22, 2026

    • LMhdX-Duct-Optimization

      Public
      Duct-optimization studies run on LMhdX: study code, run records, figures, tables and write-ups
      Python
      •
      MIT License
      •1•0•0•0•Updated Oct 20, 2026Oct 20, 2026

    • RadiShPICR

      Public
      Radially electrostatic and spherically symmetric particle-in-cell with relativity
      particle-in-cellplasma-physicsnumerical-relativity
      particle-in-cellplasma-physicsnumerical-relativityblackholesblackhole-simulationmadison-wisconsin
      Python
      •1•1•0•0•Updated Oct 19, 2026Oct 19, 2026

    • virtual_casing_jax

      Public
      Virtual Casing Calculation for Stellarator/Tokamak Magnetic Fields
      Python
      •
      Apache License 2.0
      •1•1•0•2•Updated Oct 16, 2026Oct 16, 2026

    • booz_xform_jax

      Public
      JAX version of STELLOPT's BOOZ_XFORM code
      Python
      •
      MIT License
      •1•7•0•1•Updated Oct 16, 2026Oct 16, 2026

    • JAX-in-Cell

      Public
      1D3V Particle in Cell Code in JAX
      particle-in-cell1dplasmas
      particle-in-cell1dplasmas
      Python
      •
      MIT License
      •11•34•2•10•Updated Oct 15, 2026Oct 15, 2026

    • 1D_Vlasov_JAX_Solver

      Public
      This repository provides a numerical solver for the one-dimensional Vlasov-Poisson equation using JAX for high-performance computing. The solver leverages GPU a…
      Python
      •
      MIT License
      •0•2•0•0•Updated Oct 6, 2026Oct 6, 2026

    • flexipde

      Public
      Flexible physics solver
      Python
      •
      MIT License
      •0•1•0•0•Updated Oct 6, 2026Oct 6, 2026

    • Dark-JAX-in-Cell

      Public
      A haunted, differentiable Maxwell-Proca companion to JAX-in-Cell
      particle-in-cellplasma-physicsjax
      particle-in-cellplasma-physicsjaxdark-photonmaxwell-proca
      Python
      •
      MIT License
      •0•0•0•0•Updated Oct 3, 2026Oct 3, 2026

    • pyQSC_JAX

      Public
      Python
      •
      MIT License
      •0•1•0•2•Updated Oct 3, 2026Oct 3, 2026

    • PIC_AGENTIC_WORKFLOW

      Public
      Safe agentic drift-optimization workflows around JAX-in-Cell
      Python
      •1•4•3•1•Updated Oct 1, 2026Oct 1, 2026

    • MHX

      Public
      Python
      •
      MIT License
      •0•1•0•4•Updated Sep 28, 2026Sep 28, 2026

    • magpylib_jax

      Public
      Differentiable JAX-native magnetic field library inspired by Magpylib
      Python
      •
      BSD 2-Clause "Simplified" License
      •0•3•0•0•Updated Jul 19, 2026Jul 19, 2026

    • ML_based_Coil_Optimization

      Public
      Jupyter Notebook
      •0•0•0•0•Updated May 19, 2026May 19, 2026

    • linear_gk_jax

      Public
      Python
      •
      MIT License
      •0•1•0•0•Updated Apr 29, 2026Apr 29, 2026

    • .github

      Public
      0•0•0•0•Updated Apr 10, 2026Apr 10, 2026
    ProTip! When viewing an organization's repositories, you can use the props. filter to filter by custom property.