stackav-oss / dltypeLinks
Deep Learning Type Library
☆36Updated this week
Alternatives and similar repositories for dltype
Users that are interested in dltype are comparing it to the libraries listed below
Sorting:
- FID computation in Jax/Flax.☆29Updated last year
- Clockwork VAEs in JAX/Flax☆32Updated 4 years ago
- Lightweight ML Experiment Logging 📖☆80Updated last year
- Easily run PyTorch on multiple GPUs & machines☆57Updated 3 weeks ago
- Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)☆52Updated this week
- Image augmentation library for Jax☆41Updated last year
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆47Updated last year
- Fast dataset format and loader☆23Updated 3 weeks ago
- Lightweight Experiment & Resource Monitoring 📺☆187Updated 2 years ago
- Train vision models using JAX and 🤗 transformers☆100Updated last month
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- ☆31Updated last week
- TorchFix - a linter for PyTorch-using code with autofix support☆152Updated 5 months ago
- AdaCat☆49Updated 3 years ago
- WandB sweeps integration with Hydra sweeper☆50Updated 2 years ago
- DiffuLab is designed to provide a simple and flexible way to train diffusion models while allowing full customization of its core compone…☆42Updated 3 weeks ago
- A simple, performant and scalable JAX-based world modeling codebase.☆123Updated 2 weeks ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆39Updated 4 years ago
- Document parameters using comments☆10Updated 4 years ago
- GPT implementation in Flax☆18Updated 4 years ago
- Parallel data preprocessing for NLP and ML.