Pinned Loading

1. 3-body-simulation 3-body-simulation Public

   Small simulator for N body problems

   C++ 1

2. CUDA-maskedConvolutions CUDA-maskedConvolutions Public

   Custom CUDA kernel to accelerate masked 3D convolutions

   Cuda 2

3. General-Physics-Transformer General-Physics-Transformer Public

   A foundation model to learn multiple physical systems at once

   Python 29 1

4. ML-Training-Suite ML-Training-Suite Public

   Generic training suite for ML models. Includes lots of boiler code (compile, DDP, checkpointing etc)

   Python 1 1