IDRIS-CNRS / DLO-JZLinks
formation Deep Learning Optimisé pour Jean Zay
☆16Updated 3 months ago
Alternatives and similar repositories for DLO-JZ
Users that are interested in DLO-JZ are comparing it to the libraries listed below
Sorting:
- ☆350Updated last week
- Agustinus' very opiniated publication-ready plotting library☆69Updated 4 months ago
- Compositional Linear Algebra☆491Updated last month
- SC24 Deep Learning at Scale Tutorial Material☆33Updated 7 months ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆286Updated last week
- Running Jax in PyTorch Lightning☆113Updated 9 months ago
- CLU lets you write beautiful training loops in JAX.☆355Updated 3 months ago
- TUI for the Slurm Workload Manager☆209Updated last month
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆16Updated this week
- Collaborative documentation for and from Jean Zay users. Official Jean Zay documentation: http://www.idris.fr/eng/jean-zay/☆126Updated last year
- JAX Arrays for human consumption☆106Updated 2 months ago
- ☆233Updated 7 months ago
- JMP is a Mixed Precision library for JAX.☆208Updated 7 months ago
- ☆281Updated last year
- Public repo for course material on Bayesian machine learning at ENS Paris-Saclay and Univ Lille☆90Updated 7 months ago
- Orbax provides common checkpointing and persistence utilities for JAX users☆427Updated this week
- Material for the SC22 Deep Learning at Scale Tutorial☆41Updated 2 years ago
- Library for reading and processing ML training data.☆539Updated this week
- JAX Synergistic Memory Inspector☆179Updated last year
- A projection-based framework for gradient-free and parallel learning☆104Updated 3 months ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆665Updated this week
- ☆454Updated 11 months ago
- Pytorch-like dataloaders for JAX.☆94Updated 4 months ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆207Updated last month
- Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.☆59Updated 3 years ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆496Updated last week
- JAX-Toolbox☆340Updated this week
- Deep Learning, an Energy Approach☆210Updated 3 months ago
- Notes for the Numerics of Machine Learning Lecture Course at the University of Tübingen☆215Updated last year
- Run PyTorch in JAX. 🤝☆290Updated 3 weeks ago