NVIDIA / TorchFort
An Online Deep Learning Interface for HPC programs on NVIDIA GPUs
☆149Updated this week
Related projects: ⓘ
- This repository containts materials for End-to-End AI for Science☆98Updated 3 weeks ago
- Pytorch bindings for Fortran☆83Updated 8 months ago
- XLB: Accelerated Lattice Boltzmann (XLB) based on JAX for Physics-based ML☆188Updated last week
- Repo of optimized training recipes for accelerating PyTorch workflows of AI driven surrogates for physical systems☆54Updated 9 months ago
- NVIDIA curated collection of educational resources related to general purpose GPU programming.☆58Updated last month
- GPU accelerated multigrid library for Python☆48Updated 2 weeks ago
- Differentiable Fluid Dynamics Package☆312Updated last week
- Framework providing pythonic APIs, algorithms and utilities to be used with Modulus core to physics inform model training as well as high…☆161Updated this week
- SmartSim Infrastructure Library.☆227Updated last week
- A hands-on introduction to tuning GPU kernels using Kernel Tuner https://github.com/KernelTuner/kernel_tuner/☆25Updated last week
- Datasets and code for results presented in the ProbConserv paper☆51Updated 3 months ago
- The ALCF hosts a regular simulation, data, and learning workshop to help users scale their applications. This repository contains the exa…☆52Updated 3 months ago
- An Adaptive Pencil Decomposition Library for NVIDIA GPUs☆55Updated 2 months ago
- JAX-DIPS is a differentiable interfacial PDE solver.☆41Updated last week
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆421Updated this week
- ☆56Updated last week
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆81Updated 3 months ago
- Fortran interfaces for ROCm libraries☆68Updated this week
- SLATE is a distributed, GPU-accelerated, dense linear algebra library targetting current and upcoming high-performance computing (HPC) sy…☆84Updated 2 months ago
- ☆45Updated last year
- ALCF Computational Performance Workshop☆33Updated last year
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆63Updated this week
- Simple OpenACC Fortran Examples☆52Updated 3 years ago
- NVIDIA Math Libraries for the Python Ecosystem☆194Updated 2 months ago
- Machine Learning for HPC Workflows☆119Updated last month
- Datasets and code for results presented in the BOON paper☆41Updated last year
- Neural network based solvers for partial differential equations and inverse problems . Implementation of physics-informed neural networks…☆136Updated last year
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆112Updated 4 months ago
- A parallel programming training mini app simulating weather-like flows☆137Updated 5 months ago
- A flyweight in situ visualization and analysis runtime for multi-physics HPC simulations☆189Updated last week