xflash96 / pybind11_package_example
Example python package with pybind11 cpp extension
☆57Updated 3 years ago
Related projects: ⓘ
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆57Updated 4 years ago
- ☆64Updated 10 months ago
- Texture mapping with variational auto-encoders☆40Updated 2 years ago
- a lightweight transformer library for PyTorch☆71Updated 2 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆50Updated 3 years ago
- ☆156Updated 4 years ago
- ☆45Updated 4 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- AdaCat☆49Updated 2 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆41Updated 3 years ago
- Performant, differentiable reinforcement learning☆121Updated 3 months ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 3 years ago
- Deep Probabilistic Programming Course @ DIKU☆56Updated 4 years ago
- Very deep VAEs in JAX/Flax☆45Updated 3 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆59Updated last year
- Fast Discounted Cumulative Sums in PyTorch☆95Updated 3 years ago
- Clockwork VAEs in JAX/Flax☆31Updated 3 years ago
- Normalizing Flows in Jax☆105Updated 4 years ago
- ☆97Updated 2 months ago
- ☆49Updated 3 years ago
- ☆9Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆103Updated 2 years ago
- Pytrees + dataclasses ❤️☆60Updated 2 months ago
- Differentiable Algorithms and Algorithmic Supervision.☆101Updated last year
- Annotate python source code☆68Updated 4 years ago
- A Pytree Module system for Deep Learning in JAX☆215Updated last year
- Lightweight Cluster/Cloud VM Job Management 🚀☆40Updated 3 weeks ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆47Updated 3 years ago
- ☆85Updated 3 years ago
- Neural Turing Machines in pytorch☆46Updated 2 years ago