stackav-oss / dltypeLinks
Deep Learning Type Library
β30Updated 2 weeks ago
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β22Updated 9 months ago
- Serialize JAX, Flax, Haiku, or Objax model params with π€`safetensors`β47Updated last year
- β31Updated 3 months ago
- Lightweight ML Experiment Logging πβ80Updated last year
- FID computation in Jax/Flax.β28Updated last year
- TorchFix - a linter for PyTorch-using code with autofix supportβ149Updated last month
- Clockwork VAEs in JAX/Flaxβ32Updated 4 years ago
- AdaCatβ49Updated 3 years ago
- Gym environment for playing Wordle with RL agentsβ43Updated 3 years ago
- Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)β52Updated last year
- Generative cellular automaton-like learning environments for RL.β19Updated 8 months ago
- PyTorch centric eager mode debuggerβ48Updated 10 months ago
- β23Updated 3 years ago
- β21Updated last year
- Easily run PyTorch on multiple GPUs & machinesβ47Updated last week
- Modular, scalable library to train ML modelsβ166Updated last week
- Document parameters using commentsβ10Updated 4 years ago
- Lightweight Cluster/Cloud VM Job Management πβ42Updated last year
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'β38Updated 3 years ago
- A simple, performant and scalable JAX-based world modeling codebaseβ76Updated this week
- Image augmentation library for Jaxβ40Updated last year
- β27Updated 3 weeks ago
- WandB sweeps integration with Hydra sweeperβ50Updated last year
- πͺ Interpreto is an interpretability toolbox for LLMsβ36Updated last week
- DiffuLab is designed to provide a simple and flexible way to train diffusion models while allowing full customization of its core componeβ¦β38Updated last week
- A case study of efficient training of large language models using commodity hardware.β68Updated 3 years ago
- Write your code as tree-like expressions, then transform itβ21Updated last year
- A small library for creating and manipulating custom JAX Pytree classesβ56Updated 2 years ago
- Test pytorch code with minimal computational overheadβ26Updated 2 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"β59Updated 3 years ago