gorodnitskiy/jax-cuda-docker gorodnitskiy/jax-cuda-docker

  • Stars
    star
    6
  • Rank 2,526,163 (Top 51 %)
  • Language
    Dockerfile
  • Created over 1 year ago
  • Updated over 1 year ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
gorodnitskiy/jax-cuda-docker
github

gorodnitskiy

Reviews

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

Repository Details

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

More Repositories

1

yet-another-lightning-hydra-template

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

pytorch-infer-utils

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

coco_utils

COCO supported utils
Python
1
star
4

advanced_argparse

Parser based on argparse module and YAML config
Python
1
star
5

ocr_car_registration_plates

Jupyter Notebook
1
star
6

vocpascal_segmentation

Multiclass image segmentation for VOCPascal-2012
Jupyter Notebook
1
star
7

fish_detection

Object detection pipeline for fish class trained on Open-Images dataset
Jupyter Notebook
1
star
8

lego_segmentation

Image segmentation pipeline for Lego blocks
Jupyter Notebook
1
star
9

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
1
star