Pinned Loading

1. DifferentiableUniverseInitiative/JaxPM DifferentiableUniverseInitiative/JaxPM Public

   JAX-powered Cosmological Particle-Mesh N-body Solver

   Jupyter Notebook 33 15

2. DifferentiableUniverseInitiative/jaxDecomp DifferentiableUniverseInitiative/jaxDecomp Public

   JAX bindings for the NVIDIA cuDecomp library

   Python 26 1

3. slides slides Public

   My slides and presentations

   SCSS

4. astro-informatics/s2fft astro-informatics/s2fft Public

   Differentiable and accelerated spherical transforms with JAX

   Python 142 9

5. jax-hpc-profiler jax-hpc-profiler Public

   A convinient python script to plot performance graphs for HPC JAX codes

   Python 1

6. Slurm Scripting and Profiling Slurm Scripting and Profiling

   1

   #!/bin/bash

   2

   ##########################################

   3

   ## SELECT EITHER tkc@a100 OR tkc@v100 ##

   4

   ##########################################

   5

   #SBATCH --account glc@a100