thomasahle / tensorgrad
Tensor Network Library with Autograd
☆106Updated last week
Related projects: ⓘ
- A package for defining deep learning models using categorical algebraic expressions.☆53Updated last month
- The boundary of neural network trainability is fractal☆155Updated 7 months ago
- Multiple dispatch over abstract array types in JAX.☆100Updated 3 weeks ago
- ☆325Updated 11 months ago
- Uncertainty quantification with PyTorch☆311Updated last month
- ☆180Updated 2 months ago
- Named Tensors for Legible Deep Learning in JAX☆147Updated this week
- Compositional Linear Algebra☆391Updated 2 weeks ago
- Exact OU processes with JAX☆30Updated last month
- Run PyTorch in JAX. 🤝☆187Updated 10 months ago
- Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.☆159Updated last year
- JAX Arrays for human consumption☆87Updated last year
- Interactive textbook on state-space models☆170Updated 7 months ago
- Code for the book "The Elements of Differentiable Programming".☆57Updated 3 weeks ago
- Notebooks for the "JAX in Action" book☆88Updated 2 months ago
- A functional training loops library for JAX☆84Updated 7 months ago
- Scalable neural net training via automatic normalization in the modular norm.☆108Updated 3 weeks ago
- Neural Networks for JAX☆82Updated 2 months ago
- Loopy belief propagation for factor graphs on discrete variables in JAX☆124Updated 3 months ago
- About A collection of AWESOME things about information geometry Topics☆137Updated 2 months ago
- A Jax-based library for designing and training transformer models from scratch.☆272Updated 3 weeks ago
- ☆56Updated 2 years ago
- Implementation of Diffusion Transformer (DiT) in JAX☆246Updated 3 months ago
- ☆40Updated 2 months ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆173Updated 2 years ago
- My writings about ARC (Abstraction and Reasoning Corpus)☆55Updated 3 weeks ago
- Resources from the EleutherAI Math Reading Group☆50Updated 2 months ago
- Minimal Implementation of Bayesian Optimization in JAX☆82Updated 4 months ago
- Mathematical operations for JAX pytrees☆181Updated 4 months ago
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆74Updated 8 months ago