PrzeChoj / PierwszakNieZoltodziob
☆14Updated 8 months ago
Related projects ⓘ
Alternatives and complementary repositories for PierwszakNieZoltodziob
- ☆96Updated 3 months ago
- ☆300Updated this week
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆215Updated last week
- ☆388Updated 3 weeks ago
- A library to inspect and extract intermediate layers of PyTorch models.☆470Updated 2 years ago
- CLU lets you write beautiful training loops in JAX.☆321Updated 2 months ago
- For optimization algorithm research and development.☆417Updated this week
- ☆749Updated last month
- Curated list of interactive ML demos☆338Updated 11 months ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆332Updated 2 weeks ago
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,184Updated 2 weeks ago
- Compositional Linear Algebra☆430Updated 2 weeks ago
- Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.☆479Updated 2 weeks ago
- 100 exercises to learn JAX☆567Updated 2 years ago
- Effortless plugin and play Optimizer to cut model training costs by 50%. New optimizer that is 2x faster than Adam on LLMs.☆377Updated 5 months ago
- Puzzles for exploring transformers☆323Updated last year
- Interactive textbook on state-space models☆175Updated 9 months ago
- Unofficial JAX implementations of deep learning research papers☆151Updated 2 years ago
- Turn SymPy expressions into trainable JAX expressions.☆322Updated 6 months ago
- ☆536Updated last month
- ☆27Updated 10 months ago
- JAX Synergistic Memory Inspector☆164Updated 3 months ago
- Neural Networks library in pure numpy☆65Updated 8 months ago
- JAX - A curated list of resources https://github.com/google/jax☆1,544Updated 4 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆326Updated last month
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆246Updated this week
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆516Updated this week
- Gaussian processes in JAX.☆459Updated last week
- The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series o…☆659Updated 11 months ago
- Uncertainty quantification with PyTorch☆324Updated last week