stackav-oss / dltypeLinks
Deep Learning Type Library
☆34Updated this week
Alternatives and similar repositories for dltype
Users that are interested in dltype are comparing it to the libraries listed below
Sorting:
- Fast dataset format and loader☆23Updated 11 months ago
- WandB sweeps integration with Hydra sweeper☆50Updated last year
- Easily run PyTorch on multiple GPUs & machines☆54Updated 2 weeks ago
- Scalable Computation of Hessian Diagonals☆14Updated last year
- Code for minimum-entropy coupling.☆32Updated 3 weeks ago
- Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)☆52Updated last year
- Lightweight ML Experiment Logging 📖☆81Updated last year
- FID computation in Jax/Flax.☆29Updated last year
- Automatically take good care of your preemptible TPUs☆37Updated 2 years ago
- TorchFix - a linter for PyTorch-using code with autofix support☆152Updated 3 months ago
- Clockwork VAEs in JAX/Flax☆32Updated 4 years ago
- Image augmentation library for Jax☆40Updated last year
- Lightweight Cluster/Cloud VM Job Management 🚀☆42Updated last year
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated 10 months ago
- Implementation of Reinforce for educational purposes.☆12Updated 2 years ago
- Building blocks for productive research☆64Updated 4 months ago
- Gym environment for playing Wordle with RL agents☆43Updated 3 years ago
- ☆21Updated last year
- Train vision models using JAX and 🤗 transformers☆100Updated this week
- A collection of meta-learning algorithms in Jax☆23Updated 3 years ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆47Updated last year
- A simple, performant and scalable JAX-based world modeling codebase.☆116Updated last month
- Fast Discounted Cumulative Sums in PyTorch☆96Updated 4 years ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Updated 6 months ago
- AdaCat☆49Updated 3 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- ☆60Updated 3 years ago
- ML/DL Math and Method notes☆65Updated 2 years ago
- ☆13Updated this week
- A case study of efficient training of large language models using commodity hardware.☆68Updated 3 years ago