ASKabalan

deeplearning4astro_tools

Definitly-not-vscode

fgbuster

Parametric component separation for Cosmic Microwave Background observations

jax-hpc-profiler

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

Jax-multihost

A Tutorial for JAX Multi controller distribution

slides

My slides and presentations

s2fft

Differentiable and accelerated spherical transforms with JAX

jaxDecomp

JAX bindings for the NVIDIA cuDecomp library

JaxPM

JAX-powered Cosmological Particle-Mesh N-body Solver

ASKabalan/Jax-multihost

A Tutorial for JAX Multi controller distribution

ASKabalan/deeplearning4astro_tools

ASKabalan/Definitly-not-vscode

ASKabalan/fgbuster

Parametric component separation for Cosmic Microwave Background observations

ASKabalan/jax-hpc-profiler

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

ASKabalan/jax-pmesh

A JAX-compatible port of the pmesh library for parallel mesh-based computations with GPU acceleration and automatic differentiation support.

ASKabalan/nvim-config-2.5

ASKabalan/plinalg

SPMD Linear Algebra using JAX

ASKabalan/PlotNeuralNet

Latex code for making neural networks diagrams

ASKabalan/Tutorials

ASKabalan/slides

My slides and presentations

ASKabalan/CUDALibrarySamples

CUDA Library Samples

ASKabalan/cufft-benchmarks

ASKabalan/diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

ASKabalan/fastpm-python

Python version of FastPM

ASKabalan/furax-compsep-paper

ASKabalan/gemm-benchmarks

ASKabalan/infrared-cloud-detection

Deep-learning architecture classify and identify cloud structure on sky infrared images. Standard Convolutional Neural Network (CNN) distinguishes clear sky images from cloud images. An UNet-based segmentation model detects cloud structure onto pre-identified cloudy images and outputs a probabilistic map for each pixel.

ASKabalan/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

ASKabalan/jax-grid-search

ASKabalan/jax_cosmo

A differentiable cosmology library in JAX

ASKabalan/jaxdecomp-benchmarks

ASKabalan/JaxPM-Paper

Content for the JaxPM Paper

ASKabalan/LPTLensingComparison

Repository to exactly compare an LPT full-field lensing forward model against 2pt constraints

ASKabalan/nvhpc-pypi-workflow

ASKabalan/Particle-mesh-animations

ASKabalan/pmesh

Particle Mesh in Python

ASKabalan/s2fft

Differentiable and accelerated spherical transforms with JAX

ASKabalan/spmd-jax-template

ASKabalan/tfbft