gorodnitskiy/jax-cuda-docker

Stars
9
Rank 1,939,727 (Top 39 %)
Language
Dockerfile
Created almost 2 years ago
Updated almost 2 years ago

gorodnitskiy/jax-cuda-docker

gorodnitskiy

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

This repo contains Dockerfile that can be used to easily run JAX with CUDA support in Docker without JAX and CUDA/cuDNN versions mismatching.

yet-another-lightning-hydra-template

Flexible and scalable template based on PyTorch Lightning + Hydra. Efficient workflow and reproducibility for rapid ML experiments.

pytorch-infer-utils

This package proposes simplified exporting pytorch models to ONNX and TensorRT, and also gives some base interface for ML models inference.

coco_utils

COCO supported utils

ocr_car_registration_plates

Jupyter Notebook

vocpascal_segmentation

Multiclass image segmentation for VOCPascal-2012

Jupyter Notebook

advanced_argparse

Parser based on argparse module and YAML config

fish_detection

Object detection pipeline for fish class trained on Open-Images dataset

Jupyter Notebook

lego_segmentation

Image segmentation pipeline for Lego blocks

Jupyter Notebook

coursera_advanced_ml

This repo contains solution for 1C-Predict-Future_Sales Kaggle competition (How to win data science competition Coursera course: final project)

Jupyter Notebook