Skip to content

@uwplasma UW Plasma

All

    Repositories list

    45 repositories

    • GKX

      Public
      Gyrokinetic Spectral Solver of Plasma Turbulence
      Python
      •
      MIT License
      •1•4•0•3•Updated Sep 2, 2026Sep 2, 2026

    • vmex

      Public
      JAX Version of VMEC2000
      Python
      •
      MIT License
      •4•10•2•2•Updated Sep 2, 2026Sep 2, 2026

    • NEOPAX

      Public
      NEOPAX: Neoclassical Transport Package in JAX
      Python
      •
      MIT License
      •1•3•3•3•Updated Sep 2, 2026Sep 2, 2026

    • DKX

      Public
      JAX Version of SFINCS
      Python
      •
      MIT License
      •2•1•0•1•Updated Sep 2, 2026Sep 2, 2026

    • ESSOS

      Public
      e-Stellarator Simulation and Optimization Suite
      optimizationautogradfusion
      optimizationautogradfusionjax
      Python
      •
      MIT License
      •6•18•4•21•Updated Sep 2, 2026Sep 2, 2026

    • DRBX

      Public
      Drift-Reduced Braginskii Solver in JAX
      Python
      •
      MIT License
      •0•2•0•0•Updated Sep 2, 2026Sep 2, 2026

    • NTX

      Public
      Neoclassical Transport Code
      Python
      •
      MIT License
      •0•0•0•2•Updated Sep 2, 2026Sep 2, 2026

    • booz_xform_jax

      Public
      JAX version of STELLOPT's BOOZ_XFORM code
      Python
      •
      MIT License
      •0•6•1•0•Updated Sep 1, 2026Sep 1, 2026

    • SOLVAX

      Public
      Differentiable structured linear solvers, preconditioners and matrix-free methods in JAX
      Python
      •
      MIT License
      •1•3•1•0•Updated Aug 30, 2026Aug 30, 2026

    • LMX

      Public
      LaMinar inductionless MHD code in JAX
      Python
      •
      MIT License
      •0•0•0•0•Updated Aug 30, 2026Aug 30, 2026

    • virtual_casing_jax

      Public
      Virtual Casing Calculation for Stellarator/Tokamak Magnetic Fields
      Python
      •
      Apache License 2.0
      •0•0•0•0•Updated Aug 28, 2026Aug 28, 2026

    • JAX-in-Cell

      Public
      1D3V Particle in Cell Code in JAX
      particle-in-cell1dplasmas
      particle-in-cell1dplasmas
      Python
      •
      MIT License
      •12•32•0•5•Updated Aug 27, 2026Aug 27, 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•0•0•0•Updated Aug 26, 2026Aug 26, 2026

    • MHX

      Public
      Python
      •
      MIT License
      •0•0•0•4•Updated Aug 26, 2026Aug 26, 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
      •2•3•0•3•Updated Aug 20, 2026Aug 20, 2026

    • pyQSC_JAX

      Public
      Python
      •
      MIT License
      •0•0•0•2•Updated Aug 15, 2026Aug 15, 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•0•0•0•Updated Jul 30, 2026Jul 30, 2026

    • SPECTRAX

      Public
      Solves Vlasov-Maxwell equations by doing a Hermite-Fourier decomposition.
      Python
      •
      MIT License
      •2•4•1•30•Updated Jul 23, 2026Jul 23, 2026

    • magpylib_jax

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

    • NEO_JAX

      Public
      Calculation of epsilon effective using a JAX version of NEO from STELLOPT
      Python
      •
      MIT License
      •0•1•0•0•Updated May 26, 2026May 26, 2026

    • ML_based_Coil_Optimization

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

    • PIC_AGENTIC_WORKFLOW

      Public
      Safe agentic drift-optimization workflows around JAX-in-Cell
      Python
      •1•3•3•0•Updated May 7, 2026May 7, 2026

    • linear_gk_jax

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

    • .github

      Public
      0•0•0•0•Updated Apr 10, 2026Apr 10, 2026

    • Stellarator_Webapp

      Public
      TypeScript
      •2•0•0•2•Updated Mar 11, 2026Mar 11, 2026

    • BIMFx

      Public
      Fixed boundary solution to vacuum magnetic field equilibria using Boundary Integral Methods
      Python
      •
      MIT License
      •0•0•0•0•Updated Feb 11, 2026Feb 11, 2026

    • LX

      Public
      Laplace Equation Solver in a Torus
      Python
      •
      MIT License
      •0•1•1•0•Updated Feb 10, 2026Feb 10, 2026

    • Coil_Optimization_NeuralNet

      Public
      Python
      •0•0•0•0•Updated Feb 6, 2026Feb 6, 2026

    • torus_voltage_solver

      Public
      Python
      •
      MIT License
      •0•0•0•0•Updated Feb 3, 2026Feb 3, 2026

    • pyNACX

      Public
      Python implementation of the Quasisymmetric Stellarator Construction method using automatic differentiation libraries
      Python
      •
      MIT License
      •1•1•0•0•Updated Jan 17, 2026Jan 17, 2026
    ProTip! When viewing an organization's repositories, you can use the props. filter to filter by custom property.