xor2k / npy-append-array
Create Numpy .npy files by appending on the growth axis
☆59Updated last year
Related projects: ⓘ
- Running Jax in PyTorch Lightning☆76Updated last year
- A Pytree Module system for Deep Learning in JAX☆215Updated last year
- Documentation:☆119Updated last year
- Mathematical operations for JAX pytrees☆181Updated 4 months ago
- Multidimensional indexing for tensors☆108Updated last year
- ☆59Updated 9 months ago
- A metrics library for the JAX ecosystem☆36Updated last year
- Neural Networks for JAX☆82Updated 2 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆100Updated 2 months ago
- ☆56Updated 2 years ago
- Convert scikit-learn models to PyTorch modules☆157Updated 4 months ago
- ☆24Updated last year
- NumPy arrays, ready for human consumption☆63Updated 2 months ago
- Lightweight ML Experiment Logging 📖☆79Updated 3 weeks ago
- The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends☆74Updated this week
- Lightweight Hyperparameter Optimization 🚂☆145Updated 3 weeks ago
- Tools for JAX☆44Updated last week
- This is a port of Mistral-7B model in JAX☆29Updated 2 months ago
- Multiple dispatch over abstract array types in JAX.☆100Updated last month
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Updated last year
- Framework-agnostic library for checking array/tensor shapes at runtime.☆47Updated 3 years ago
- Keras-like APIs for JAX framework☆50Updated last year
- A functional training loops library for JAX☆84Updated 7 months ago
- A small library for creating and manipulating custom JAX Pytree classes☆59Updated last year
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆39Updated 11 months ago
- Provides symbolic API for model creation in PyTorch.☆61Updated last week
- JMP is a Mixed Precision library for JAX.☆183Updated 4 months ago
- Run PyTorch in JAX. 🤝☆187Updated 10 months ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆173Updated 2 years ago