Henrique Mendonça's repositories
FourCastNet
Initial public release of code, data, and model weights for FourCastNet
production
General interest repository for CSCS users
pytorch-examples
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
albumentations
fast image augmentation library and easy to use wrapper around other libraries
contrastive-ne
Code for parametric neighbor embeddings
crystfelparser
A python tool to parse the output of indexamajig
cscs-reframe-tests
The CSCS ReFrame test suite
ddp_starter
A most basic working setup for training ML-models with Pytorch using DDP and mulitple nodes.
docker-inbound-agents
Jenkins docker inbound agents with tools built in.
dynamic-resource-provisioning
Ansible-powered Dynamic Storage Resource Provisioning (DSRP)
e2cnn
E(2)-Equivariant CNNs Library for Pytorch
fast-feedback-indexer
Develop an indexer for fast feedback in protein crystallography experiments
faster-pytorch-blog
Outlining techniques for improving the training performance of your PyTorch model without compromising its accuracy
kubeflow-manifests
A repository for Kubeflow Kustomize manifests
llama-recipes
Examples and recipes for Llama 2 model
mpi4jax
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
stockfish
Integrates the Stockfish chess engine with Python (Official fork)
storage
MLPerf™ Storage Benchmark Suite
tensorflow
An Open Source Machine Learning Framework for Everyone
torchbraid
A layer-parallel neural network technology built on PyTorch
TransformerEngine
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.
webdataset
A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.